Python - 結合文本文件（特定行）

Question

我有兩個來自一個實驗的大型文本數據文件，我想用特殊方式將它分割成一個文件。

數據的小樣本：

文件1：

plotA 10 
plotB 9 
plotC 9 

文件2：

98% 
7/10 
21 
98% 
5/10 
20 
98% 
10/10 
21 

而且我想結果是這樣的：

plotA 10 98% 7/10 21 
plotB 9 98% 5/10 20 
plotC 9 98% 10/10 21 

我不知道它如何在python中解決。我試圖重新排序文件2具有：

lines = file2.readlines() 
aaa = lines[0] + lines[3] + lines[6] 
bbb = lines[1] + lines[4] + lines[7] 
ccc = lines[2] + lines[5] + lines[8] 

，並使用拉鍊，但我失敗了（這方法耗時大的文本文件）。

任何幫助？

Answer 1

您可以使用itertools.izip_longest切片文件2三重線，然後再次使用使用它與第一個文件來壓縮他們：

from itertools import izip_longest 
with open('file1.txt') as f1, open('file2.txt') as f2: 

    args = [iter(f2)] * 3 
    z = izip_longest(f1, izip_longest(*args), fillvalue='-') 
    for line, tup in z: 
      print '{:11}'.format(line.strip()), '{:5}{:5}{:>5}'.format(*map(str.strip, tup)) 

如果你想要這個結果寫入到一個新的文件，你可以打開文件寫入，而不是打印它寫入文件中的行。

結果：

plotA 10 98% 7/10 21 
plotB 9 98% 5/10 20 
plotC 9 98% 10/10 21 

Answer 2

下面是一個例子，你就會有錯誤處理和所有改進它：^）

file1 = open('file1') 
file2 = open('file2') 

# take one line in file1 
for line in file1: 
     # print result with tabulation to separate fields 
     print '\t'.join(
       # the line from file1 
       [line.strip()] + 
       # and three lines from file2 
       [file2.readline().strip() for _ in '123'] 
     )  

請注意，我使用的字符串'123'因爲它比range(3)短（並且不需要函數調用）;它必須是任何產生三個步驟的迭代器。

只讀取所需的數據並進行處理，避免了在內存中加載所有文件的需要（正如您所說的文件很大）。

乾杯。