So finden Sie den Index einer Liste in einer Liste

Question

Grundsätzlich habe ich eine TXT-Datei, die eine Ausgabe von einem Fortran-Modell ist. Die Ausgabe sieht ein wenig wie folgt aus:

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

Jeder Schritt ist ein weiterer Satz, wie das oben so schließlich ich eine TXT-Datei mit 3000+ Schritte haben.

Also möchte ich alle Temperaturen bei jedem Schritt abrufen. Ich versuche etwas zu schreiben, um alle Punkte zu indizieren, wo 'Temp (K)' erscheint und füge dann 1 zu diesem Index hinzu, um den Ort der tatsächlichen Temperatur zu erhalten.

Mein Code sieht wie folgt aus:

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 

So offensichtlich ist dies nicht der eleganteste Weg, aber ich bin sehr viel in den tiefen Ende mit Python. Wie finde ich den Index einer Liste in der Liste (main2), wenn der erste Wert dieser Liste == 'Temp (K)'?

Nb. Ich benutze np und matplot, um die Daten danach zu plotten.

Answer 1

die Datei in eine Liste lesen.

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

die Nummern der Linien erhalten, die eine Temperatur (unter Verwendung einer Liste Verständnis, weil es schnell ist) beginnen:

die Temperaturen holen, wieder mit einer Liste Verständnis.

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

Falls gewünscht, können Sie auch diese verbinden sich zu einer Liste Verständnis:

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

Ein Beispiel in 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] 

Answer 2

Dies kann mit Pythons enumerate() Funktion durchgeführt werden.

Hier ist ein Beispiel für das Extrahieren der Indizes. (! Und, nur zum Spaß, die Temperaturen zu)

idxs = [] 
tempature_values = [] 
for idx, value in enumerate(main2): 
    # Check if value is not empty & the first element is 'Temp(K)' 
    if value and value[0] == 'Temp(K)': 
     idxs.append(idx) 
     temp_values.append(main2[idx+1][0]) 

Answer 3

Wie wäre es damit:

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)