Question

基本上我有一个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文件。

所以我想回想一下每一步的温度。我正在尝试写一些东西来索引所有的点，其中温度（K）＆＃39;出现，然后将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。那么如果列表的第一个值==＆＃39; Temp（K）＆＃39;？

，如何在列表（main2）中找到列表的索引？

的Nb。我在这之后用np和matplot绘制数据。

Answer 1

这可以使用python的enumerate()函数来完成。

这是提取索引的一个例子。（而且，只是为了好玩，温度太高了！）

ids = User.joins(:memberships)
          .where(membership_packet_sent_at: nil)
          .select(&:current_membership_packet_expired?)
          .uniq
          .map(&:id)

User.where(id: ids)

Answer 2

这个怎么样：

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)

Answer 3

将文件读入列表。

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

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

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

获得温度，再次列表理解。

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)')]

IPython中的示例：

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]

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

3 个答案: