pandas读写excel、csv

读/写excel

二维数组写入excel

import pandas as pd

# 整一个5*5的二维数组
list_2d= [['哦豁']*5]*5
# 转成pandas的dataframe
df = pd.DataFrame(list_2d,columns=['第一列', '第二列', '第三列', '第四列', '第五列'], index=['第一行', '第二行', '第三行', '第四行', '第五行'])
# 写入excel
df.to_excel(r'D:./source.xlsx')

读取刚才写入的excel

import pandas as pd

# 读excel的时候会自动把第一行的每个元素作为每一列的列名，但是不会自动把第一列的元素作为每一行的行名，因此需要加入index_col=0参数
df = pd.read_excel(r'D:./source.xlsx', index_col=0)

---

读/写csv

读csv和读excel类似，只是改一下方法名和保存文件路径中的扩展名

import pandas as pd

df = pd.read_csv(r'D:/source.csv', index_col=0)
df = pd.to_csv(r'D:/source.csv')

---