Pandas模块的DataFrame

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()