如何從文件中讀取從任意換行開始線在Python

Question

我在那個線與新線分離的方式格式化的文件，像下面

1 1 1 1 
2 2 2 2 2 2 2 
3 3 3 3 3 3 

我想單獨看線例如從第二個開始，並將它們保存在一個數組中。我想我可以管理最後一部分，但我無法弄清楚如何從文件的第n個換行開始讀取。 任何想法，我該怎麼做？ 謝謝。 此致敬禮。

Answer 1

嗯，你可以做這樣的事情：

n1, n2 = 0, 2 

with open('filename.txt') as f: 
    print '\n'.join(f.read().split('\n')[n1:n2+1]) 

這將產生（按照內容的文件在您發佈）輸出像這樣：

1 1 1 1 
2 2 2 2 2 2 2 
3 3 3 3 3 3 

EDIT 1：

@mi c-tiz根據你在下面發表的評論，我知道你希望將文本文件中的所有數字都放到一個數組中。

with open('filename.txt') as f: 
    array = [i for i in f.read() if not i == ' '] 

你提到這個代碼，會產生一個列表array

array = ['1', '1', '1', '1', '\n', '2', '2', '2', '2', '2', '2', '2', '\n', '3', '3', '3', '3', '3', '3'] 

然後，您可以通過拆分它\n字符的打印出現的元素。

編輯2： 您可以使用下面

d = {} 
with open('filename.txt') as f: 
    array = f.read().split('\n') 

for i in range(len(array)): 
    d['l%r'%i] = [int(j) for j in array[i] if not j == ' '] 

代碼在字典中保存這些數字這將產生d = {'l2': [3, 3, 3, 3, 3, 3], 'l0': [1, 1, 1, 1], 'l1': [2, 2, 2, 2, 2, 2, 2]}

Answer 2

由於文件是可迭代的Python中，你可以調用它的next跳過第一行，例如：

with open('data.txt', 'r') as data: 
    next(data) 
    for line in data: 
     print line.split() 

會產生：

['2', '2', '2', '2', '2', '2', '2'] 
['3', '3', '3', '3', '3', '3'] 

參考文獻：

• next
• str.split

Answer 3

lines = open('test.txt', 'r').readlines() 

# n is your desired line 
for lineno in range(n-1, len(lines)): 
    print list(lines[lineno].strip()) 

Answer 4

不能直接跳轉到指定的行。你必須閱讀前n行：

n = 1 
with open('data.txt', 'r') as data: 
    for idx, _ in enumerate(data): 
     if idx == n: 
      break 
    for line in data: 
     print line.split() 

Answer 5

您可以使用itertool.islice對於這一點，如：

from itertools import islice 

with open('filename') as fin: 
    wanted = islice(fin, 1, None) # change 1 to lines to skip 
    data = [line.split() for line in wanted]