python编写自动化脚本处理excel文件

说明：通过以下四步可以实现文件的读取、查询、修改、保存操作。

工具：pandas

import pandas as pd

1.读取文件

readfile = pd.read_excel(r"文件名.xlsx")

2.数据查询

def querydata(datafile,data):
    '''查询函数，输入查询文件和查询数据，返回需要的数据位置'''
    if data in datafile.values:
        key = datafile.columns[datafile.eq(data).any(axis=0)]
        index = datafile.index[datafile.eq(data).any(axis=1)].tolist()
        return key[0],index[0]

补充说明：file.columns是excel的列名，file.index是每列的元素索引

举例：该文件中a,h,f,d,e,d,r即列名

若要对查询到的数据的列名进行修改：

# 获取列名key在readfile.columns中的位置
key_position = list(readfile.columns).index(key)
# 获取该列后一列的数据
revise_position = readfile.columns[key_position + 1]

3.数据修改

readfile[列名][索引]=123

列名为str数据类型，索引为int数据类型，类似字典

4.文件保存

readfile.to_excel('revise_3.xlsx', index=False)