Python 使用xlsxwriter模块读取Excel文件 使用pandas对数据进行筛选保存将第一行全部数据存到新的Excel文件第一行中 第二行只保留用户账号,姓名,所属组织存到新的Excel文件中
时间: 2024-02-01 07:13:24 浏览: 69
使用pandas进行excel文件的读取写入
5星 · 资源好评率100%
以下是使用xlsxwriter和pandas模块读取Excel文件并保存数据的示例代码:
```python
import pandas as pd
import xlsxwriter
# 读取Excel文件
df = pd.read_excel('input_file.xlsx')
# 将第一行全部数据存到新的Excel文件第一行中
workbook = xlsxwriter.Workbook('output_file.xlsx')
worksheet = workbook.add_worksheet()
for col_num, value in enumerate(df.columns.values):
worksheet.write(0, col_num, value)
# 第二行只保留用户账号,姓名,所属组织存到新的Excel文件中
filtered_df = df[['用户账号', '姓名', '所属组织']]
filtered_df.to_excel('output_file.xlsx', startrow=1, index=False, header=False)
workbook.close()
```
在上面的代码中,我们首先使用pandas模块读取Excel文件,并将第一行的数据存储到新的Excel文件的第一行中。然后,我们使用pandas的数据筛选功能将需要的数据存储到一个新的数据框架中,并将其存储到新的Excel文件的第二行中。最后,我们使用xlsxwriter模块创建新的Excel文件并将数据写入其中,并关闭文件。
阅读全文