在文件中将字符串匹配后保存n行到变量

Question

我正在尝试编写一个可以读取文本文件并将特定行另存为列表的脚本。文本文件如下所示：

TRIAL1

能量：54432

坐标：

0.7 0.4 0.5

0.3 0.4 0.1

0.3 0.4 0.3

文件中还有更多试验（TRIAL2，TRIAL3等，具有相应的能量和坐标），但是我希望能够选择一个特定的试验并将其对应的坐标保存在列表中（以供以后操作）。 / p>

到目前为止，我已经写过：

with open('energy.txt') as f:
   for line in f:
        if 'TRIAL1' in line:

我不确定如何跳过“能量”和“坐标”行，而仅将坐标保存到变量中。

Answer 1

假定“ TRIAL”块都与您的示例相同（以适应您的脚本）：

print(df['Timestamp'])

0         1552154111
1         1552154115
2         1552154117
3         1552154120
4         1552154125
5         1552154127
6         1552154134
7         1552154137