Python Excel

安装 Python 库

pip install openpyxl pandas

代码实现示例

创建 Excel 文件

import pandas as pd

df = pd.DataFrame(
    {"ID": [1, 2, 3], "name": ["张三", "李四", "王五"], "age": [20, 21, 22]}
)  # 创建一个DataFrame对象
print(df)
df = df.set_index("ID")  # 将ID列设置为索引
print(df)
df.to_excel("test.xlsx")  # 将DataFrame对象写入到Excel文件中
print("Excel 文件已创建成功!")

读取 Excel 文件

import pandas as pd

df = pd.read_excel("test.xlsx", engine="openpyxl")  # 读取Excel文件
print(df)
print(f"类型为:{type(df)}")  # 打印DataFrame对象的类型
print(df.shape)  # 打印DataFrame对象的形状(行数,列数)
print(df.columns)  # 打印DataFrame对象的列名
print(df.head())  # 打印DataFrame对象的前5行

修改 Excel 文件

import pandas as pd

df = pd.read_excel("test.xlsx")  # 读取Excel文件
print(df)
df.drop(1, inplace=True)  # 删除ID为1的行
print(df)
df.to_excel("test.xlsx", index=False)  # 将DataFrame对象写入到Excel文件中

# 删除多行
df = pd.read_excel("test.xlsx")  # 读取Excel文件
print(df)
df.drop([1, 2], inplace=True)  # 删除ID为1和2的行
print(df)
df.to_excel("test.xlsx", index=False)  # 将DataFrame对象写入到Excel文件中

# 删除多行但无法同时进行时,应该先删除行号大的行
data = [1, 3, 5, 7, 9]
df = pd.read_excel("test.xlsx")  # 读取Excel文件
print(df)
for i in reversed(data):  # reversed()函数将列表倒序
    df.drop(i, inplace=True)  # 删除ID为1,3,5,7,9的行
print(df)