如何在Python中对N行文本文件进行交互

时间:2016-11-02 17:50:21

标签: python-2.7 python-3.x parsing split multiple-columns

您好我正在尝试转换文件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

到disp.mol:

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 ...
  

我用open打开了disp.txt文件('disp.txt','w')作为f:

     

使用lines = f.readlines()

读入行      

在行中使用for x分割数据:numbers = x.split()

     

然后   vib1.append(float(numbers [2])),vib2.append并存储它们   vib1 = [],vib2 = []等。

当我需要将我以格式存储的数据放在disp.mol中时,我的问题就出现了。使用下面的代码,我能够从前两行获得前三次振动的输出,但我不确定如何在接下来的两行中执行相同的循环(以及如果我有更多的2N行) 。我也不确定如何对每个振动进行编号。任何帮助都将不胜感激。

with open('disp.mol', 'w') as thisfile:
        thisfile.writelines('vibration')
        thisfile.writelines('\n')
        for x in range (0, 2):
                vib_one = str(vib1[x]) + '  ' +  str(vib2[x]) + '  ' +  str(vib3[x])
                thisfile.writelines(vib_one)
                thisfile.writelines('\n')
        thisfile.writelines('vibration')
        thisfile.writelines('\n')
        for x in range (0, 2):
                vib_two = str(vib4[x]) + '  ' +  str(vib5[x]) + '  ' +  str(vib6[x])
                thisfile.writelines(vib_two)
                thisfile.writelines('\n')
        thisfile.writelines('vibration')
        thisfile.writelines('\n')
        for x in range (0, 2):
                vib_three = str(vib7[x]) + '  ' +  str(vib8[x]) + '  ' +  str(vib9[x])
                thisfile.writelines(vib_three)
                thisfile.writelines('\n')

输出:

vibration
0.12  -0.91  0.39
0.0  -0.02  0.0
vibration
-0.4  0.31  0.85
-0.05  0.05  0.12
vibration
-0.66  -0.18  -0.22
-0.57  -0.26  -0.29

1 个答案:

答案 0 :(得分:0)

这是一种方法:

with open('disp.txt') as f, open('disp.mol','w') as out:
    vibration = 1
    for line in f:
        line1 = line.split()
        line2 = next(f).split() # also get next line
        for i in range(2,len(line1),3):
            out.write('vibration {}\n'.format(vibration))
            out.write(' '.join(line1[i:i+3])+'\n')
            out.write(' '.join(line2[i:i+3])+'\n')
            vibration += 1

输出:

vibration 1
0.12 -0.91 0.39
0.00 -0.02 0.00
vibration 2
-0.40 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
0.01 0.09 0.19
0.47 -0.33 -0.79
vibration 6
-0.71 -0.21 -0.26
0.57 0.16 0.19