pythonexcel文件读写操作类封装
时间: 2023-07-26 10:23:48 浏览: 166
python关于excel类的封装
以下是一个简单的 Python Excel 文件读写操作的类封装示例:
```python
import pandas as pd
class ExcelFile:
def __init__(self, file_path):
self.file_path = file_path
self.data = None
# 读取 Excel 文件
def read_file(self):
try:
# 使用 pandas 库读取 Excel 文件
self.data = pd.read_excel(self.file_path)
except FileNotFoundError:
print(f"File {self.file_path} not found.")
# 写入 Excel 文件
def write_file(self):
try:
# 将数据写入到 Excel 文件中
writer = pd.ExcelWriter(self.file_path)
self.data.to_excel(writer, index=False)
writer.save()
print(f"File {self.file_path} saved successfully.")
except:
print(f"Failed to save file {self.file_path}.")
# 测试代码
if __name__ == "__main__":
# 创建 ExcelFile 对象
excel_file = ExcelFile("example.xlsx")
# 读取 Excel 文件
excel_file.read_file()
if excel_file.data is not None:
print(excel_file.data)
# 写入 Excel 文件
new_data = {"Name": ["John", "Mary", "Peter"], "Age": [30, 25, 40]}
df = pd.DataFrame(new_data)
excel_file.data = df
excel_file.write_file()
```
在这个示例中,我们定义了一个名为 `ExcelFile` 的类,这个类封装了 Excel 文件的读取和写入操作。在初始化函数中,我们传入一个文件路径,并将其保存为类的属性。`read_file` 函数尝试读取这个文件,并将读取到的数据保存为类的属性。`write_file` 函数将类的属性中的数据写入到对应的文件中。
在测试代码中,我们首先创建一个 `ExcelFile` 对象,并调用 `read_file` 函数来读取一个名为 `example.xlsx` 的 Excel 文件,并将读取到的数据输出到控制台。然后我们创建一个新的 DataFrame 对象,并将其保存为 `ExcelFile` 对象的属性,最后调用 `write_file` 函数将这个 DataFrame 中的数据写入到一个名为 `example.xlsx` 的 Excel 文件中。
阅读全文