python - 将多个列表写入单个列

Question

我想读取这样的 2 列文件：

1 3
4 7
2 8

并取回这样的单列文件：

3
7
7
7
7
8
8

这是我的代码：

import csv
array1 = []
array2 = []
with open('test.txt', 'rb') as f:
    reader = csv.reader(f, delimiter=' ')
    for row in reader:
        array1.append(row[0])
        array2.append(row[1])
range1 = len(array1)
array1 = [int(x) for x in array1]
array2 = [int(x) for x in array2]

outfile = open('drp', 'wb')
writer=csv.writer(outfile)

for s in range(0, range1, 1):
            res = [array2[s]] * array1[s]
            print res
            writer.writerow([res])
outfile.close()  

除了最后的写作步骤外，它似乎正在工作。我在输出文件中得到的是：

[3]  
"[7, 7, 7, 7]"  
"[8, 8]"

显然我没有正确使用 csv 编写器。有什么建议么？谢谢。

Answer 1

你把这个简单的问题复杂化了：

import csv
with open('test.txt') as f, open('out.csv','w') as f2:
    writer = csv.writer(f2, delimiter = '\n')
    for line in f:
        x, y = line.split() 
        writer.writerow([y]*int(x))

输出：

>>> cat out.csv
3
7
7
7
7
8
8

Answer 2

您以二进制模式或使用 csv 读取文本文件有什么特殊原因吗？

with open('test.txt', 'r') as f, open('drp','w') as ff::
     for l in f:
         a,b=l.split(' ')
         for i in range(int(a)):
             ff.write(b+'\n')

Answer 3

另外的选择

from itertools import chain
with open('test.txt') as f:
    z = (line.split() for line in f)
    print '\n'.join(chain.from_iterable(int(x)*[y] for (x, y) in z))

Answer 4

这应该这样做

with open('path/to/input') as infile, open('path/to/output', 'w') as outfile:
    for line in infile:
        mult, val = line.strip().split()
        outfile.write('\n'.join([val for _ in xrange(int(mult))]))
        outfile.write('\n')