以下是使用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()

请确保在运行脚本时提供正确的文件名作为参数

写一个python脚本使用pandas模块用集合实现从参数1获取文件名文件内容做为查询条件将同results-20230616T1300csv的id列一致的行打印id列和text列

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

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