以下是使用 Pandas 模块编写的 Python 脚本,根据参数 1 中的文件名,将每一行作为查询条件,打印与 "results-20230616T1300.csv" 中的 "id" 列匹配的行的 "id" 列和 "text" 列:\n\npython\nimport pandas as pd\nimport sys\n\n# 从参数 1 获取文件名\nfilename = sys.argv[1]\n\n# 读取查询条件文件\nquery_conditions = pd.read_csv(filename, header=None)\n\n# 读取结果文件\nresults = pd.read_csv('results-20230616T1300.csv')\n\n# 使用集合实现查询条件与结果文件的匹配\nquery_results = results[results['id'].isin(set(query_conditions[0]))]\n\n# 打印匹配的结果\nprint(query_results[['id', 'text']])\n\n\n请确保将脚本保存为 .py 文件,并在命令行中运行,将参数 1 替换为查询条件文件的路径。

Python 脚本:使用 Pandas 和集合从 CSV 文件中提取匹配数据

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

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