python提取Excel部分字段数据

python提取Excel部分字段数据

说明

python读取Excel文件数据，通过指定的header，即字段，提取对应的字段数据，生成新的Excel文件。

文件格式转换

另存Excel为csv格式，csv以纯文本形式存储表格数据。

安装依赖

pip install csv

code

提取aaa.csv文件中字段为xxx和yyy的两列数据，过滤其他数据，并生成新的bbb.csv文件。

import csv

# 待处理文件，修改
file = 'aaa.csv'

# 输出文件，修改
file_out = 'bbb.csv'

# 输出文件的字段名，修改
header = ['xxx', 'yyy']

# 输出文件写header
f_out = open(file_out, 'w', newline='')
writer = csv.DictWriter(f_out, header)
writer.writeheader()

# 输出文件写内容
with open(file, 'r') as f:
    reader = csv.DictReader(f)
    for row in reader:
        # 写数据到输出文件，修改
        writer.writerow({'xxx': row['xxx'], 'yyy': row['yyy']})

f_out.close()