import pandas as pd import sys

从参数1获取文件名

filename = sys.argv[1]

读取查询条件文件

with open(filename, 'r') as file: query_conditions = file.readlines() query_conditions = [condition.strip() for condition in query_conditions]

读取results-20230616T1300.csv文件

#df = pd.read_csv('results-20230616T1300.csv',encoding='GBK') df = pd.read_csv('items-updated.csv',encoding='GBK')

使用集合实现查询并打印结果

for condition in query_conditions: result = df[df['ITEM_ID'] == condition] if not result.empty: print(result[['ITEM_ID', 'DESCRIPTION']].to_string(index=False))

Python 脚本优化:去除多余表头,打印 Pandas DataFrame 结果

原文地址: https://www.cveoy.top/t/topic/pWuT 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录