import pandas as pd\nimport sys\n\n# 获取文件名参数\nfile_name = sys.argv[1]\n\n# 读取查询条件文件\nquery_df = pd.read_csv(file_name)\n\n# 读取结果文件\nresult_df = pd.read_csv('results-20230616T1300.csv')\n\n# 使用集合实现查询\nquery_set = set(query_df['id'])\nresult_set = set(result_df['id'])\n\n# 打印匹配的行\nmatched_rows = result_set.intersection(query_set)\nfor row_idx in matched_rows:\n print("id:", result_df.loc[row_idx, 'id'])\n print("text:", result_df.loc[row_idx, 'text'])\n print()\n\n请确保在运行脚本时提供正确的文件名作为参数。

Python Pandas 脚本:使用集合实现文件内容匹配并打印结果

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

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