我有一个包含许多列的文件"TAB.csv".我想从CSV文件中选择一个没有标题的列(该列的索引是3).然后创建一个新的文本文件"NEW.txt"并在那里写入该列(不带标题).
下面的代码读取该列,但带有标题.如何省略标题并将该列保存在新的文本文件中?
import csv
with open('TAB.csv','rb') as f:
reader = csv.reader(f)
for row in reader:
print row[3]
Run Code Online (Sandbox Code Playgroud)
ave*_*net 12
这是@tmrlvi正在讨论的解决方案:它通过下一个函数跳过第一行(标题):
import csv
with open('TAB.csv','rb') as input_file:
reader = csv.reader(input_file)
output_file = open('output.csv','w')
next(reader, None)
for row in reader:
row_str = row[3]
output_file.write(row_str + '\n')
output_file.close()
Run Code Online (Sandbox Code Playgroud)
试试这个:
import csv
with open('TAB.csv', 'rb') as f, open('out.txt', 'wb') as g:
reader = csv.reader(f)
reader.next() # skip header
g.writelines(row[3] + '\n' for row in reader)
Run Code Online (Sandbox Code Playgroud)