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-with-quotes.csv',encoding='GBK') df = pd.read_csv('interactions-data-with-likes_rocket-qat.csv',encoding='GBK')

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

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