DataFrame
# 操作DataFrame类似于操作字典
import pandas as pd
# 创建DataFrame对象
df = pd.DataFrame([[100, 100], [96, 90], [93, 100]])
df.index = ["小明", "小王", "小张"] # 设置行索引
df.columns = ["语文", "数学"] # 设置列索引
df1 = pd.DataFrame([[100, 100], [96, 90]], index=["小明", "小王"], columns=["语文", "数学"])
df2 = pd.DataFrame({"语文": [100, 96], "数学": [100, 90]}, index=["小明", "小王"])
file = "./demo1.xlsx"
writer = pd.ExcelWriter(file)
df.to_excel(writer, "sheet1")
df1.to_excel(writer, "sheet2")
df2.to_excel(writer, "sheet3")
# 访问元素
print("小明:\n", df.loc["小明"]) # 提取一行数据,返回series对象
print("小明、小张:\n",df.loc[["小明", "小张"]]) # 提取两行数据
print("语文:\n", df.loc[:, "语文"]) # 提取一列数据
print("语文:\n", df.语文) # 提取一列数据
print("第一行:\n", df.iloc[0]) # 按索引值(整数)提取一行数据
print("第一列:\n", df.iloc[:, 0]) # 按索引值(整数)提取一列数据
# print(df.head()) # 显示头部部分数据
# print(df.tail()) # 显示尾部部分数据
# 修改、增删元素
df.loc["小明", "语文"] = 95 # 修改单个元素值
df.loc[:, "数学"] += 10 # 修改一列元素值
df.loc["小杨"] = [200, 200] # 增加一行数据
df.loc[:, "英语"] = [80, 80, 80, 80] # 增加一列数据
df.insert(0, "政治", [100, 100, 100, 100]) # 在第一列插入一列
df3 = df.drop("小张", axis=0) # 删除一行,返回一个新DataFrame
df4 = df.drop("政治", axis=1) # 删除一列
df.to_excel(writer, "sheet4")
df3.to_excel(writer, "sheet5")
df4.to_excel(writer, "sheet6")
writer.close()