"使用Python脚本,利用Pandas库,从文本文件获取查询条件,并在CSV文件中查找匹配的'id'列,并打印匹配结果的'id'列和'text'列。脚本无需循环,代码简洁高效。\n\npython\nimport sys\nimport pandas as pd\n\ndef main():\n # 获取文件名参数\n filename = sys.argv[1]\n\n # 读取查询条件文件\n with open(filename, 'r') as file:\n queries = file.readlines()\n\n # 去除查询条件中的换行符\n queries = [query.strip() for query in queries]\n\n # 读取目标文件\n target_file = 'results-20230616T1300.csv'\n df = pd.read_csv(target_file)\n\n # 使用查询条件进行筛选\n filtered_df = df[df['id'].isin(queries)]\n\n # 打印筛选结果的'id'列和'text'列\n print(filtered_df[['id', 'text']])\n\nif __name__ == "__main__":\n main()\n\n\n在命令行中运行该脚本时,需要提供一个文本文件作为参数1,例如:\n\nbash\npython script.py conditions.txt\n\n\n其中,conditions.txt是包含查询条件的文本文件。脚本会读取该文件的每一行作为查询条件,然后在results-20230616T1300.csv文件中查找与查询条件的'id'列一致的行,最后打印匹配结果的'id'列和'text'列。

Python脚本:使用Pandas从文本文件查询CSV数据并输出匹配结果

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

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