写一个python脚本使用pandas模块用集合实现从参数1获取文件名文件内容做为查询条件将同results-20230616T1300csv的id列一致的行打印id列和text列
以下是使用pandas模块编写的Python脚本,实现了根据文件名作为查询条件,打印与results-20230616T1300.csv文件中的"id"列匹配的行的"id"列和"text"列:
import pandas as pd
import sys
# 获取文件名参数
file_name = sys.argv[1]
# 读取查询条件文件
query_df = pd.read_csv(file_name)
# 读取结果文件
result_df = pd.read_csv('results-20230616T1300.csv')
# 使用集合实现查询
query_set = set(query_df['id'])
result_set = set(result_df['id'])
# 打印匹配的行
matched_rows = result_set.intersection(query_set)
for row_idx in matched_rows:
print("id:", result_df.loc[row_idx, 'id'])
print("text:", result_df.loc[row_idx, 'text'])
print()
请确保在运行脚本时提供正确的文件名作为参数
原文地址: https://www.cveoy.top/t/topic/idPz 著作权归作者所有。请勿转载和采集!