python - 将多行文本文件拆分为多行 csv 文件

Question

我有一个文本文件，其中包含以下形式的数据；

100157  100157
100157  364207
100157  38848
100157  bradshaw97introduction
100157  bylund99coordinating
100157  dix01metaagent
100157  gray99finding
...
...

我正在尝试使用以下方法将其转换为 scikit 可读数据集:

datafile = open(filename.txt, 'r')
data=[]
for row in datafile:
    data.append(row.strip().split('\t'))

c1 = open(filename.csv, 'w')
arr = str(data)
c.write(arr)
c.close

但是在执行此代码后，数据会输出到一行中，而我打算将数据以 csv 格式整齐地按行和列分开，就像 Iris 数据集那样。

我能得到一些关于我应该如何进行的帮助吗？谢谢。

Answer 1

使用csv module :

import csv

with open('filename.txt', 'r') as f, open('filename.csv', 'w') as fout:
    writer = csv.writer(fout)
    writer.writerows(line.rstrip().split('\t') for line in f)

输出csv文件:

100157,100157
100157,364207
100157,38848
100157,bradshaw97introduction
100157,bylund99coordinating
100157,dix01metaagent
100157,gray99finding
...