Python 如何在列表中查找列表的索引

基本上我有一个txt文件，它是fortran模型的输出。输出看起来有点像这样：

Title:Model

Temp(K)    Ionic str    Rho    Phi    H2O    Ice ...
273.15     4            1.003  1.21   1000   0.00

Species    Ini Conc    Final Conc     Act ....
H          0.0         0.12032        0.59
NH4        3.0         3.00           0.43
Cl         1.0         1.00           0.47
...

Title:Model

Temp(K)    Ionic str    Rho    Phi    H2O    Ice ...
273.15     4            1.003  1.21   1000   0.00

Species    Ini Conc    Final Conc     Act ....
H          0.0         0.12032        0.59
NH4        3.0         3.00           0.43
Cl         1.0         1.00           0.47
...

每一步都会添加另一组，如上面所述，因此最终我有一个包含3000多个步骤的txt文件

所以我想回忆一下每一步的温度。我试图写一些东西来索引所有出现“Temp（K）”的点，然后在索引中加上1来获得实际温度的位置

我的代码如下所示：

import numpy as np
import matplotlib.pyplot as plt
main=[]
main2=[]
count=0
with open('FrOut.txt', 'r') as f:
    data=f.readlines()
    for line in data:
        main.append(line.split(','))
for value in main:
    for x in value:
        main2.append(x.split())
for value in main2:
    for x in value:
            if x=='Temp(K)':count+=1

很明显，这不是最优雅的方式，但我对python非常熟悉。那么，如果列表（main2）的第一个值=='Temp（K）'，我如何在列表（main2）中找到列表的索引呢

---

注意。在此之后，我将使用np和matplot来绘制数据。

这可以使用python函数来完成

下面是一个提取索引的示例。（还有，只是为了好玩，还有温度！）

这个怎么样：

temp_rows = []

with open(f, 'rb') as fin:
    reader = csv.reader(fin)
    for index, row in enumerate(reader):
        if 'Temp(K)' in [word for words in row for word in words.split()]:
            temp_rows.append(index)

---

将文件读入列表

with open('FrOut.txt', 'r') as f:
    data=f.readlines()

获取开始温度的行数（使用列表理解，因为它很快）：

获取温度，再次使用列表

temps = [float(data[n].split()[0]) for n in idx]

如果需要，您甚至可以将其合并为一个列表：

temps = [float(data[n+1].split()[0]) for n, ln in enumerate(data) if ln.startswith('Temp(K)')]

例如：

---

您在文本上循环了两次，在第二个列表中进行索引查找。您可以只使用

enumerate（）

，只循环一次在第二个理解中，我循环索引，而不是整个行列表。我真的可以用一种理解力就能做到。在本例中，为了清晰起见，我选择将其拆分。正如我所说的，索引查找

enumerate（）

once比使用

enumerate（）

as-is的两个列表理解更清楚。谢谢，这就是我想要的！

temps = [float(data[n].split()[0]) for n in idx]

temps = [float(data[n+1].split()[0]) for n, ln in enumerate(data) if ln.startswith('Temp(K)')]

In [1]: text = '''Title:Model
   ...: 
   ...: Temp(K)    Ionic str    Rho    Phi    H2O    Ice ...
   ...: 273.15     4            1.003  1.21   1000   0.00
   ...: 
   ...: Species    Ini Conc    Final Conc     Act ....
   ...: H          0.0         0.12032        0.59
   ...: NH4        3.0         3.00           0.43
   ...: Cl         1.0         1.00           0.47
   ...: ...
   ...: 
   ...: Title:Model
   ...: 
   ...: Temp(K)    Ionic str    Rho    Phi    H2O    Ice ...
   ...: 273.15     4            1.003  1.21   1000   0.00
   ...: 
   ...: Species    Ini Conc    Final Conc     Act ....
   ...: H          0.0         0.12032        0.59
   ...: NH4        3.0         3.00           0.43
   ...: Cl         1.0         1.00           0.47
   ...: ...
   ...: '''

In [2]: data = text.splitlines()

In [3]: idx = [n+1 for n, ln in enumerate(data) if ln.startswith('Temp(K)')]

In [4]: idx
Out[4]: [3, 14]

In [5]: [float(data[n].split()[0]) for n in idx]
Out[5]: [273.15, 273.15]

In [6]: [float(data[n+1].split()[0]) for n, ln in enumerate(data) if ln.startswith('Temp(K)')]
Out[6]: [273.15, 273.15]