读取Excel
import pandas as pd
file = "./demo.xlsx"
df = pd.read_excel(file, sheet_name="Sheet1") # 默认第一行作为表头,表头即为列索引,返回DataFrame
data = df.values # 提取DataFrame的值,返回ndarray,不含表头
print('df:\n', df)
print('data:\n', data)
df = pd.read_excel(file, header=1, index_col=0) # header指定第2行为表头,即列索引,index_col指定第1列为行索引
data = df.values # 返回数组不含表头及以上的行,也不含作为行索引的第1列
print('df:\n', df)
print('data:\n', data)
写入Excel
import numpy as np
import pandas as pd
a = np.array([[100, 100], [96, 90], [93, 100]])
file = "./demo1.xlsx"
writer = pd.ExcelWriter(file)
name = ["小明", "小王", "小张"]
course = ["语文", "数学"]
df = pd.DataFrame(a, index=name, columns=course) # columns为表头,列索引;index为行索引
df.to_excel(writer, "sheet1") # 写入表头和行索引
# df.to_excel(writer, "sheet1", index=None, header=None) # 不写入表头和行索引
writer.close()