在转到下一个循环之前，在for循环中遍历N行

Question

我在How to interate over N lines of a text file in Python上发布了一个问题，但我想让程序更适用于N行(例如，我想一次收集四到六行，而不是一次收集两行)。

我想把这个disp.txt：

116  C             0.12 -0.91  0.39    -0.40  0.31  0.85    -0.66 -0.18 -0.22
117  O             0.00 -0.02  0.00    -0.05  0.05  0.12    -0.57 -0.26 -0.29
116  C            -0.03 -0.04  0.00     0.01  0.09  0.19    -0.71 -0.21 -0.26
117  O            -0.14  0.88 -0.45     0.47 -0.33 -0.79     0.57  0.16  0.19

要这样做：

vibration 1
0.12  -0.91  0.39
0.0  -0.02  0.0
vibration 2
-0.4  0.31  0.85
-0.05  0.05  0.12
vibration 3
-0.66  -0.18  -0.22
-0.57  -0.26  -0.29
vibration 4 
-0.03 -0.04 0.00 
-0.14 0.88 -0.45
vibration 5 ...

下面是我到目前为止所得到的

vibration = 1
with open('disp1.txt','r') as f, open('disp2.txt', 'w') as out:

 while True:
  next_lines = list(islice(f,2))
  m = next_lines
  for i in m:
   s = i.strip().split('\n')
   for j in s:
    p = j.split()
    v = len(p)
    for u in range(2,5,3):
     out.write('vibration {}\n'.format(vibration))
     out.write(' '.join(p[u:u+3])+'\n')
     vibration += 1
    for x in range(5,8,3):
     out.write(' '.join(p[x:x+3])+'\n')
    for y in range(8,11,3):
      out.write(' '.join(p[y:y+3])+'\n')
  if not next_lines:
   break

它显示了以下内容：

vibration 1
0.12 -0.91 0.39
-0.40 0.31 0.85
-0.66 -0.18 -0.22
vibration 2
0.00 -0.02 0.00
-0.05 0.05 0.12
-0.57 -0.26 -0.29
vibration 3
-0.03 -0.04 0.00
0.01 0.09 0.19
-0.71 -0.21 -0.26
vibration 4
-0.14 0.88 -0.45
0.47 -0.33 -0.79
0.57 0.16 0.19

因此，它是跨页面读取，而不是读取前三列数字的前两行，然后进入下一组三列数字，依此类推。通过使用isplice，它一次读取两行代码，但是由于我不理解的原因，这两行代码在进入下一个循环(对于x在范围(5，8，3)：)之前，都没有通过第一个循环(对于范围(2，5，3)：中的u)，而是每一行都通过每个循环。有人能帮我这个忙吗？

谢谢。

Answer 1

这应该可以做你想要的事情。

def group(lst, size = 3):
    """
    returns a list of lists where the inner lists are at most
    of length size

    [1, 2, 3, 4, 5, 6] -> [[1, 2, 3], [4, 5, 6]]
    group(list, int) -> list<list>
    """
    return [lst[i:i+size] for i in range(0, len(lst), size)]

def get_vibrations(lines):
    for vibration in zip(*[group(line) for line in lines]):
        yield '\n'.join(' '.join(v) for v in vibration)

def main(n):
    with open('in.txt', 'r') as fin, open('out.txt', 'w') as fout:
        i = 0
        for line in fin:
            line = line.strip().split()[2:]
            vibrations = [line] + [fin.readline().strip().split()[2:] for _ in range(n - 1)]

            # write each vibration to the file
            for i, v in enumerate(get_vibrations(vibrations), start = i + 1):
                fout.write('vibration {}\n{}\n'.format(i, v))

if __name__ == '__main__':
    main(4) # the number of lines per vibration

# Input
116  C             0.12 -0.91  0.39    -0.40  0.31  0.85    -0.66 -0.18 -0.22
117  O             0.00 -0.02  0.00    -0.05  0.05  0.12    -0.57 -0.26 -0.29
116  C            -0.03 -0.04  0.00     0.01  0.09  0.19    -0.71 -0.21 -0.26
117  O            -0.14  0.88 -0.45     0.47 -0.33 -0.79     0.57  0.16  0.19
116  C             0.26 -0.40 -0.83     0.26  0.31  0.25     0.33  0.71 -0.51
117  O            -0.19 -0.39 -0.36    -0.40  0.87  0.49     0.11  0.26 -0.57
116  C            -0.35  0.27  0.79    -0.03 -0.71 -0.18     0.93 -0.89 -0.81
117  O            -0.84  0.99  0.38    -0.27 -0.33  0.55     0.31  0.99 -0.18
116  C             0.76 -0.77  0.71     0.19 -0.39  0.46     0.30 -0.15  0.71
117  O            -0.97  0.31  0.58    -0.42  0.16 -0.33    -0.39 -0.17 -0.54

# Output
vibration 1
0.12 -0.91 0.39
0.00 -0.02 0.00
-0.03 -0.04 0.00
-0.14 0.88 -0.45
vibration 2
-0.40 0.31 0.85
-0.05 0.05 0.12
0.01 0.09 0.19
0.47 -0.33 -0.79
vibration 3
-0.66 -0.18 -0.22
-0.57 -0.26 -0.29
-0.71 -0.21 -0.26
0.57 0.16 0.19
vibration 4
0.26 -0.40 -0.83
-0.19 -0.39 -0.36
-0.35 0.27 0.79
-0.84 0.99 0.38
vibration 5
0.26 0.31 0.25
-0.40 0.87 0.49
-0.03 -0.71 -0.18
-0.27 -0.33 0.55
vibration 6
0.33 0.71 -0.51
0.11 0.26 -0.57
0.93 -0.89 -0.81
0.31 0.99 -0.18

如果您还想允许没有足够的行，那么您可以添加

from itertools import zip_longest

并更改为以下内容

def get_vibrations(lines):
    for vibration in zip_longest(*[group(line) for line in lines], fillvalue = ['null']):
        yield '\n'.join(' '.join(v) for v in vibration)

# Outputs | added to above output
vibration 7
0.76 -0.77 0.71
-0.97 0.31 0.58
null
null
vibration 8
0.19 -0.39 0.46
-0.42 0.16 -0.33
null
null
vibration 9
0.30 -0.15 0.71
-0.39 -0.17 -0.54
null
null

Answer 2

info = '''0.12 -0.91  0.39    -0.40  0.31  0.85    -0.66 -0.18 -0.22    0.00 -0.02  0.00    -0.05  0.05  0.12    -0.57 -0.26 -0.29    -0.03 -0.04  0.00    0.01  0.09  0.19    -0.71 -0.21 -0.26    -0.14  0.88 -0.45    0.47 -0.33 -0.79     0.57  0.16  0.19'''

new_list_1 = info.split('    ')
vib_1 = new_list_1[0] + '\n' + new_list_1[3]
vib_1 = 'vibration 1' + '\n' + vib_1
print vib_1

new_list_2 = info.split('    ')
vib_2 = new_list_2[1] + '\n' + new_list_2[4]
vib_2 = 'vibration 2' + '\n' + vib_2
print vib_2

new_list_3 = info.split('    ')
vib_3 = new_list_3[2] + '\n' + new_list_3[5]
vib_3 = 'vibration 3' + '\n' + vib_3
print vib_3