写一个python脚本用集合实现从参数1获取文件名文件内容做为查询条件将同results-20230616T1300csv的id列一致的行打印id列和text列
以下是使用集合实现的Python脚本:
import sys
# 获取文件名参数
filename = sys.argv[1]
# 读取文件内容
with open(filename, 'r') as file:
file_content = file.read()
# 将文件内容按行分割,并去除首尾空格
lines = file_content.split('\n')
lines = [line.strip() for line in lines]
# 获取查询条件
query_set = set(lines)
# 打开results-20230616T1300.csv文件并读取内容
with open('results-20230616T1300.csv', 'r') as result_file:
result_content = result_file.read()
# 将结果文件内容按行分割,并去除首尾空格
result_lines = result_content.split('\n')
result_lines = [line.strip() for line in result_lines]
# 遍历结果文件的每一行
for line in result_lines:
# 将每一行按逗号分割,并去除首尾空格
values = line.split(',')
values = [value.strip() for value in values]
# 判断当前行的id列是否在查询条件集合中
if values[0] in query_set:
# 打印id列和text列
print(values[0], values[1])
你可以将以上代码保存为一个Python脚本文件,例如script.py,然后在命令行中运行以下命令来执行脚本并传递文件名参数:
python script.py filename.txt
其中filename.txt是你想要作为查询条件的文件名。脚本将会读取该文件的内容作为查询条件,然后在当前目录下查找名为results-20230616T1300.csv的文件,并打印出与查询条件中的id列一致的行的id列和text列
原文地址: https://www.cveoy.top/t/topic/idPw 著作权归作者所有。请勿转载和采集!