Verwenden itertools.islice
,
import itertools
import math
filename = 'test.dat'
N = 9
M = 4
num_rest_lines = N
nrof_lists = int(math.ceil(N*1.0/M))
with open(filename, 'r') as f:
for i in range(nrof_lists):
num_lines = min(num_rest_lines, M)
lines_gen = itertools.islice(f, num_lines)
l = [int(line.rstrip()) for line in lines_gen]
num_rest_lines = num_rest_lines - M
print(l)
# Output
[14, 3, 21, 37]
[48, 12, 4, 6]
[22]
Vorherige Antwort: Iterate über eine Datei (N
Zeilen) in Chunks (jede M
Zeilen), bilden eine Liste von N/M+1
Listen.
import itertools
def grouper(iterable, n, fillvalue=None):
"""iterate in chunks"""
args = [iter(iterable)] * n
return itertools.izip_longest(*args, fillvalue=fillvalue)
# Test
filename = 'test.dat'
m = 4
fillvalue = '0'
with open(filename, 'r') as f:
lists = [[int(item.rstrip()) for item in chuck] for chuck in grouper(f, m, fillvalue=fillvalue)]
print(lists)
# Output
[[14, 3, 21, 37], [48, 12, 4, 6], [22, 4, 0, 0]]
Haben Sie versucht, es mit zwei 'for' Schleifen zu implementieren? Das wäre ein guter Anfang, und wenn Sie Schwierigkeiten haben, können Sie um Hilfe bitten. – amiller27
Ich habe meine Antwort bearbeitet. Bitte überprüfen. – SparkAndShine