我正在尝试使用 Python 解决以下问题:我有一个表 (csv),我想从中连续提取第 1 + 2 行,然后是第 1 + 3 行,然后是第 1 + 4 行,等等。提取物应再次保存为 csv 文件,并根据要提取的第二行中的第一个字符串命名 (2, 3, 4, 5, ...)。现在我的问题 - python 是执行此操作的正确工具吗?是否有可用的示例代码?
非常感谢您的帮助和提示!
克劳德·S.
+ 说明:
感谢您的反馈 - 实际上我正在尝试使用 csv 模块打开并使用以下代码读取表格:
import csv
import sys
f = open(sys.argv[1], 'rt')
try:
reader = csv.reader(f)
for row in reader:
print row
现在我不确定如何选择和写入所需的行...对于初学者这类问题感到抱歉。 克劳德·S.
请您参考如下方法:
fname = argv[1]
with open(fname) as i:
reader = csv.reader(i)
first_row = next(reader)
for cur_row in reader:
out_name = cur_row[0]
with open(out_name, 'wb') as o:
writer = csv.writer(o)
writer.writerow(first_row)
writer.writerow(cur_row)
希望对你有帮助
