openpyxl读取excel指定多列数据并生成新的excel
时间: 2024-01-24 17:02:10 浏览: 147
python使用openpyxl操作excel的方法步骤
可以使用openpyxl库来读取Excel文件并生成新的Excel文件。
下面是一个示例代码,可以读取指定的多列数据,并将其写入新的Excel文件中:
```
import openpyxl
# 打开原始Excel文件
workbook = openpyxl.load_workbook('input.xlsx')
# 选择工作表
worksheet = workbook['Sheet1']
# 选择要读取的列
columns = ['A', 'C', 'D']
# 创建新的Excel文件
new_workbook = openpyxl.Workbook()
# 选择要写入的工作表
new_worksheet = new_workbook.active
# 写入表头
header = ['Column A', 'Column C', 'Column D']
new_worksheet.append(header)
# 读取数据并写入新文件中
for row in worksheet.iter_rows(min_row=2, values_only=True):
data = [row[worksheet.column_dimensions[column].index] for column in columns]
new_worksheet.append(data)
# 保存新文件
new_workbook.save('output.xlsx')
```
在这个示例中,首先打开原始Excel文件。然后选择要读取的工作表和列。接下来,创建一个新的Excel文件并选择要写入的工作表。然后写入表头,并使用`iter_rows`方法读取每一行数据,并将指定的列写入新文件中。最后,保存新文件。
阅读全文