pandas读写Excel文件_pandas 读写 excel

pandas是基于numpy的一个数据分析python包，pandas读取Excel文件需要导入pandas包

import pandas as pd
1

下面简单记录pandas对excel的读写操作，以便后续查看。源码可参考：Github
现有文件example.xlxs如下

1.pandas读Excel文件

读取excel文件，并将读入的数据转换为DataFrame格式

# 读取excel
data = pd.read_excel(r'example.xlsx', sheet_name='Sheet1')
1
2

去除表格中某些列

# 去除“序号”
data = data[['姓名', '性别', '年龄', '身高', '体重']]
# 或者如下语句也可去除“序号”
data2 = data.filter(items=['姓名', '性别', '年龄', '身高', '体重'])
1
2
3
4

选择表格中某些行

# 选择“性别为男”的行
data = data[data['性别'].apply(lambda x: x == '男')]
1
2

遍历表格

# 以行遍历
for index, row in data.iterrows():
    print('姓名：{0}，性别：{1}'.format(row['姓名'], row['性别']))
1
2
3

2.pandas写Excel文件

定义写入的表格标题(columns)，也可省略(如果省略，则自动添加数字替代)；然后将代写入的数据转换为DataFrame数据结构，最后调用to_excel函数写入excel文件

# 写excel
write_data = []
for index, row in data.iterrows():
    write_data.append([row['姓名'], row['性别'], '是' if int(row['年龄']) > 25 else '否'])

# 标题
columns = ['姓名', '性别', '是否大于25岁']
# 转换为DataFrame格式
df = pd.DataFrame(write_data, columns=columns)
# 写入Excel文件
with pd.ExcelWriter('example2.xlsx') as writer:
    df.to_excel(writer, index=False, sheet_name='pandas create')
1
2
3
4
5
6
7
8
9
10
11
12