写一个python脚本使用pandas模块用集合实现从参数1获取文件名文件的每一行做为查询条件将同results-20230616T1300csv的id列一致的行打印id列和text列
以下是使用pandas模块编写的Python脚本,根据参数1中的文件名,将每一行作为查询条件,打印与results-20230616T1300.csv中的“id”列匹配的行的“id”列和“text”列:
import pandas as pd
import sys
# 从参数1获取文件名
filename = sys.argv[1]
# 读取查询条件文件
query_conditions = pd.read_csv(filename, header=None)
# 读取结果文件
results = pd.read_csv('results-20230616T1300.csv')
# 使用集合实现查询条件与结果文件的匹配
query_results = results[results['id'].isin(set(query_conditions[0]))]
# 打印匹配的结果
print(query_results[['id', 'text']])
请确保将脚本保存为.py文件,并在命令行中运行,将参数1替换为查询条件文件的路径。
原文地址: http://www.cveoy.top/t/topic/idPD 著作权归作者所有。请勿转载和采集!