Python 自动化教程(2) : Excel自动化：使用pandas库，相关源码和文件

import pandas as pd import openpyxl # DataFrame 基本操作 # read DataFrame from excel df = pd.read_excel('202201报表.xlsx') print(type(df)) print(df) # 指定worksheet的名称，读入数据表 df = pd.read_excel('202201报表.xlsx', sheet_name="客户") print(df) # df.columns 是一个 Series 对象 print(df.columns) # 显示每一列的名称 for i in range(len(df.columns)): print(df.columns[i]) # 如果存在名为"城市"的列 if '城市' in df.columns: column = df['城市'] # 取得名为"城市"的一列 print(type(column)) # column 是一个 Series 对象 print(column) print(column[0]) # 取得 columns 第1个数据（索引号从0开始） print(column.to_list()) # 将series转换为 list # 取得一行 (Series) if len(df) > 0: row = df.iloc[0] # 取得第一行(从0开始) print(type(row)) # row 是一个 Series 对象 print(row) print(row['城市']) # 取得该行中 列名为"城市"的单元格的值 # 取得一个格子的值 print(df.loc[1, '城市']) # 写入一个格子的值 df.loc[1, '城市'] = "新城" print(df.loc[1, '城市']) # 取得 金额列的 最大、最小、平均值 print(df['金额'].max(), df['金额'].min(), df['金额'].mean()) print(len(df)) # len(df)取得行数 print(len(df.columns)) # len(df.columns)取得列数 # 将DataFrame存入excel文件（注意:直接使用 to_excel() 将会丢弃原excel文件中的其他worksheet) filename = '新报表.xlsx' df.to_excel(filename, sheet_name="商品销售") def save_to_exist_excel(df, filename, sheet_name, index=False, startrow=0, startcol=0, **kwargs): """ 将DataFrame存入已有的excel文件 :param df: DataFrame 对象 :param filename: Excel 文件名 :param sheet_name: 工作表名称(标题) :param index: (可选)索引列是否保存 :param startrow: (可选)开始写入的行号, 第一行为0, 第二行为1, ... :param startcol: (可选)开始写入的列号， A列为0， B列为1 ... :param kwargs: (可选)其他参数 :return: None """ book = openpyxl.load_workbook(filename) writer = pd.ExcelWriter(filename, engine='openpyxl') writer.book = book writer.sheets = {} for ws in book.worksheets: writer.sheets[ws.title] = ws df.to_excel(writer, sheet_name=sheet_name, index=index, startcol=0, startrow=0, **kwargs) writer.save() # 读出数据 filename = '202201报表.xlsx' df = pd.read_excel('202201报表.xlsx', sheet_name="商品销售") df.loc[1, '城市'] = "新城" # 修改单元格 # 将DataFrame存入已存在的excel文件 save_to_exist_excel(df, filename, sheet_name="商品销售") # Pandas Official Document Introduction: # https://pandas.pydata.org/pandas-docs/stable/getting_started/intro_tutorials/index.html