python代码 excel文件管理系统,创建一个excel文件并进行增删改查操作
时间: 2023-07-26 18:46:32 浏览: 61
好的,下面是一个Python代码示例,用于创建Excel文件并进行增删改查操作:
```python
import openpyxl
class ExcelFileManager:
def __init__(self, file_path):
self.file_path = file_path
self.workbook = openpyxl.Workbook()
def add_sheet(self, sheet_name):
self.workbook.create_sheet(sheet_name)
def remove_sheet(self, sheet_name):
self.workbook.remove(self.workbook[sheet_name])
def read_sheet(self, sheet_name):
sheet = self.workbook[sheet_name]
data = []
for row in sheet.iter_rows(min_row=2):
row_data = []
for cell in row:
row_data.append(cell.value)
data.append(row_data)
return data
def write_sheet(self, sheet_name, data):
if sheet_name not in self.workbook.sheetnames:
self.workbook.create_sheet(sheet_name)
sheet = self.workbook[sheet_name]
for row_data in data:
sheet.append(row_data)
self.workbook.save(self.file_path)
if __name__ == '__main__':
# 创建一个Excel文件
file_manager = ExcelFileManager('example.xlsx')
# 添加一个工作表
file_manager.add_sheet('Sheet1')
# 往Sheet1中写入数据
data = [['姓名', '年龄', '性别'], ['张三', 18, '男'], ['李四', 20, '女']]
file_manager.write_sheet('Sheet1', data)
# 读取Sheet1中的数据
data = file_manager.read_sheet('Sheet1')
print(data)
# 删除Sheet1工作表
file_manager.remove_sheet('Sheet1')
```
这个类可以创建一个新的Excel文件,并进行增删改查操作。你可以使用这个类来创建Excel文件、添加工作表、写入数据、读取数据和删除工作表。在示例代码中,我们创建了一个名为`example.xlsx`的Excel文件,并向其中添加了一个名为`Sheet1`的工作表,然后写入了一些数据。接着读取了`Sheet1`中的数据并打印输出,最后删除了`Sheet1`工作表。