Extrahieren Sie Zeilen zwischen Überschriften, die durch Datei

Question

wiederholt werden. Ich versuche, eine TXT-Datei mit ~ 43 k Zeilen zu ändern. Nachdem der Befehl * Nset in der Datei angegeben wurde, muss ich alle Zeilen extrahieren und speichern, die auf diesen Befehl folgen, und stoppe, wenn der nächste * Befehl in der Datei erreicht wird. Nach jedem Befehl gibt es eine unterschiedliche Anzahl von Zeilen und Zeichen. Zum Beispiel, hier ist ein Beispiel Teil der Datei:

*Nset 

1, 2, 3, 4, 5, 6, 7, 

12, 13, 14, 15, 16, 

17, 52, 75, 86, 92, 

90, 91, 92 93, 94, 95.... 

*NEXT COMMAND 

blah blah blah 

*Nset 

numbers 

*Nset 

numbers 

*Command 

irrelevant text 

Der Code, den ich zur Zeit Werke haben, wenn die Zahlen, die ich brauche nicht zwischen zwei * Nset ist sind. Wenn ein * Nset den Zahlen eines anderen folgt, überspringt es diesen Befehl und die vorgehenden Zeilen alle zusammen und ich kann nicht herausfinden warum. Wenn der nächste Befehl nicht * Nset ist, findet er den nächsten und zieht die Daten vollkommen in Ordnung.

import re 

# read in the input deck 
deck_name = 'master.txt' 
deck = open(deck_name,'r') 

#initialize variables 
nset_data = [] 
matched_nset_lines = [] 
nset_count = 0 

for line in deck: 
    # loop to extract all nset names and node numbers 
    important_line = re.search(r'\*Nset,.*',line) 
    if important_line : 
     line_value = important_line.group() #name for nset 
     matched_nset_lines.insert(nset_count,line_value) #name for nset 
     temp = [] 

     # read lines from the found match up until the next *command 
     for line_x in deck : 
      if not re.match(r'\*',line_x): 
       temp.append(line_x) 
      else : 
       break 

     nset_data.append(temp) 

    nset_count = nset_count + 1 

Ich benutze Python 3.5. Danke für jede Hilfe.

Answer 1

Wenn Sie nur sollten die Linien zwischen *Nsets der folgende Ansatz extrahieren möchten arbeiten:

In [5]: with open("master.txt") as f: 
    ...:  data = [] 
    ...:  gather = False 
    ...:  for line in f: 
    ...:   line = line.strip() 
    ...:   if line.startswith("*Nset"): 
    ...:    gather = True 
    ...:   elif line.startswith("*"): 
    ...:    gather = False 
    ...:   elif line and gather: 
    ...:    data.append(line) 
    ...: 

In [6]: data 
Out[6]: 
['1, 2, 3, 4, 5, 6, 7,', 
'12, 13, 14, 15, 16,', 
'17, 52, 75, 86, 92,', 
'90, 91, 92 93, 94, 95....', 
'numbers', 
'numbers'] 

Und, wenn Sie weitere Informationen wünschen, ist es einfach genug, um die oben zu erweitern:

In [7]: with open("master.txt") as f: 
    ...:  nset_lines = [] 
    ...:  nset_count = 0 
    ...:  data = [] 
    ...:  gather = False 
    ...:  for i, line in enumerate(f): 
    ...:   line = line.strip() 
    ...:   if line.startswith("*Nset"): 
    ...:    gather = True 
    ...:    nset_lines.append(i) 
    ...:    nset_count += 1 
    ...:   elif line.startswith("*"): 
    ...:    gather = False 
    ...:   elif line and gather: 
    ...:    data.append(line) 
    ...: 

In [8]: nset_lines 
Out[8]: [0, 14, 18] 

In [9]: nset_count 
Out[9]: 3 

In [10]: data 
Out[10]: 
['1, 2, 3, 4, 5, 6, 7,', 
'12, 13, 14, 15, 16,', 
'17, 52, 75, 86, 92,', 
'90, 91, 92 93, 94, 95....', 
'numbers', 
'numbers'] 

Answer 2

Das macht was Sie wollen.

command = [] 
commandLines = [] 

with open('test.txt') as file: 
    for line in file: 
     if line.startswith('*'): 
      command.append(line.rstrip()) 
      commandLines.append([]) 
     else: 
      commandLines[-1].append(line.rstrip()) 

import pprint 

pprint.pprint(command) 
pprint.pprint(commandLines) 

commandLines[i] ist eine Liste entsprechende Leitungen zu command[i] enthält.

Drucke für Befehle aus:

['*Nset', '*NEXT COMMAND', '*Nset', '*Nset', '*Command'] 

und die Linien (verschachtelte Liste):

[['1, 2, 3, 4, 5, 6, 7,', 
    '12, 13, 14, 15, 16,', 
    '17, 52, 75, 86, 92,', 
    '90, 91, 92 93, 94, 95....'], 
['blah blah blah'], 
['numbers'], 
['numbers'], 
['irrelevant text']] 

Annahme: Nur Kommandozeilen beginnen mit '*'.