Python 脚本：使用集合从 CSV 文件中查找匹配项

{ "title": "Python 脚本：使用集合从 CSV 文件中查找匹配项", "description": "使用 Python 脚本，利用集合数据结构从给定文件获取查询条件，并在 "results-20230616T1300.csv" 文件中查找匹配的 ID 列和 TEXT 列，并打印结果。", "keywords": "Python, 脚本, 集合, CSV 文件, 查询条件, 匹配项, ID, TEXT", "content": ""使用集合从参数 1 获取文件名，文件内容做为查询条件，将同 results-20230616T1300.csv 的 "id" 列一致的行打印 "id" 列和 "text" 列内容：\n\npython\nimport sys\n\n# 获取文件名参数\nfilename = sys.argv[1]\n\n# 读取文件内容\nwith open(filename, 'r') as file:\n file_content = file.read()\n\n# 将文件内容按行分割，并去除首尾空格\nlines = file_content.split('\n')\nlines = [line.strip() for line in lines]\n\n# 获取查询条件\nquery_set = set(lines)\n\n# 打开 results-20230616T1300.csv 文件并读取内容\nwith open('results-20230616T1300.csv', 'r') as result_file:\n result_content = result_file.read()\n\n# 将结果文件内容按行分割，并去除首尾空格\nresult_lines = result_content.split('\n')\nresult_lines = [line.strip() for line in result_lines]\n\n# 遍历结果文件的每一行\nfor line in result_lines:\n # 将每一行按逗号分割，并去除首尾空格\n values = line.split(',')\n values = [value.strip() for value in values]\n \n # 判断当前行的 id 列是否在查询条件集合中\n if values[0] in query_set:\n # 打印 id 列和 text 列\n print(values[0], values[1])\n\n\n你可以将以上代码保存为一个 Python 脚本文件，例如 script.py，然后在命令行中运行以下命令来执行脚本并传递文件名参数：\n\n\npython script.py filename.txt\n\n\n其中 filename.txt 是你想要作为查询条件的文件名。脚本将会读取该文件的内容作为查询条件，然后在当前目录下查找名为 results-20230616T1300.csv 的文件，并打印出与查询条件中的 id 列一致的行的 id 列和 text 列。\n"}