python表格文件读写操作封装代码
时间: 2023-12-16 15:06:12 浏览: 79
以下是一个简单的 Python 表格文件读写操作的封装代码示例:
```python
import pandas as pd
# 读取表格文件
def read_table_file(file_path):
try:
# 使用 pandas 库读取表格文件
data = pd.read_csv(file_path)
return data
except FileNotFoundError:
print(f"File {file_path} not found.")
return None
# 写入表格文件
def write_table_file(data, file_path):
try:
# 将数据写入到表格文件中
data.to_csv(file_path, index=False)
print(f"File {file_path} saved successfully.")
except:
print(f"Failed to save file {file_path}.")
# 测试代码
if __name__ == "__main__":
# 读取表格文件
data = read_table_file("example.csv")
if data is not None:
print(data)
# 写入表格文件
new_data = {"Name": ["John", "Mary", "Peter"], "Age": [30, 25, 40]}
df = pd.DataFrame(new_data)
write_table_file(df, "new_example.csv")
```
在这个示例中,我们使用了 pandas 库来读取和写入表格文件。`read_table_file` 函数接收一个文件路径作为参数,尝试读取这个文件并返回一个 pandas 的 DataFrame 对象。`write_table_file` 函数接收一个 DataFrame 对象和一个文件路径作为参数,将 DataFrame 中的数据写入到对应的文件中。
在测试代码中,我们首先调用 `read_table_file` 函数来读取一个名为 `example.csv` 的表格文件,并将读取到的数据输出到控制台。然后我们创建一个新的 DataFrame 对象,并调用 `write_table_file` 函数将这个 DataFrame 中的数据写入到一个名为 `new_example.csv` 的表格文件中。
阅读全文