Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/python


from numpy import *
import os

folderpath ="/home/georg/Auswertungen/out/"
filepathlist =os.listdir(folderpath)
print filepathlist

for file in filepathlist:
    if file.endswith('.out'):
       f = open(folderpath + file, 'r')
       count_row = 0 

       for row in f:
           count_row = count_row + 1
           row_string_list = row.split() 
           length = len(row_string_list)
    
           if row_string_list[0:4] == ['Summary', 'of', 'CHF', 'Data']:
              row_number_CHF_Data = count_row
#              print row_number_CHF_Data 
#              print row_string_list 
       f.close()

       f = open(folderpath+'/'+file, 'r')
       r = open('test.csv', 'w')
       count_row = 0 

       for row in f:
           count_row = count_row + 1
           row_string_list = row.split() 
           length = len(row_string_list)
        
           if count_row == row_number_CHF_Data + 81:
              MDNBR_B =row_string_list[len(row_string_list)-3]
              Heat_Flux_B=row_string_list[len(row_string_list)-4]
              Axial_Location_B=row_string_list[len(row_string_list)-5]
              Rod_B=row_string_list[len(row_string_list)-2]
              Subchannel_B=row_string_list[len(row_string_list)-1]
#             print row_string_list
#              print MDNBR_B
              print Heat_Flux_B
#              r.writelines(Heat_Flux_B)
#             print Axial_Location_B
#             print Rod_B
#             print Subchannel_B
           if count_row == row_number_CHF_Data + 93:
              MDNBR_A=row_string_list[len(row_string_list)-3]
              Heat_Flux_A=row_string_list[len(row_string_list)-4]
              Axial_Location_A=row_string_list[len(row_string_list)-5]
              Rod_A=row_string_list[len(row_string_list)-2]
              Subchannel_A=row_string_list[len(row_string_list)-1]
#             print row_string_list
#              print MDNBR_A
#             print Heat_Flux_A
#             print Axial_Location_A
#             print Rod_A
#             print Subchannel_A
       r.write(str(Heat_Flux_B))
       f.close()
       r.close()

Code: Alles auswählen

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import csv
import os
from glob import glob


def main():
    folder_path = '/home/georg/Auswertungen/out/'
    # 
    # Offsets of the two lines of interest relative to the start of
    # the CHF data.
    # 
    offset_a, offset_b = 81, 93
    # 
    # Offsets of MDNBR, heat flux, axial location, rod, and subchannel
    # within a row.
    # 
    value_offsets = [-3, -4, -5, -2, -1]
    
    with open('test.csv', 'w') as result_file:
        writer = csv.writer(result_file, delimiter=';')
        for path in glob(os.path.join(folder_path, '*.out')):
            # 
            # Get start line of the last CHF data summery in the file.
            # 
            with open(path, 'r') as lines:
                chf_data_offset = None
                for i, line in enumerate(lines):
                    if line.startswith('Summary of CHF Data'):
                        chf_data_offset = i
                assert chf_data_offset is not None, 'No CHF data found.'
            # 
            # Extract the two rows and write them into the result CSV file.
            # 
            with open(path, 'r') as lines:
                offsets = [
                    chf_data_offset + offset_a, chf_data_offset + offset_b
                ]
                result = list()
                for i, line in enumerate(lines):
                    if i in offsets:
                        row = line.split()
                        result.append([row[n] for n in value_offsets])
                    elif i > offsets[-1]:
                        break
                assert len(result) == 2, 'Premature end of file'
                writer.writerows(result)


if __name__ == '__main__':
    main()

Code: Alles auswählen

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import csv
import os
from glob import glob
from itertools import dropwhile, islice
from operator import itemgetter


def main():
    folder_path = '/home/georg/Auswertungen/out/'
    # 
    # Offsets of the two lines of interest relative to the start of
    # the CHF data.
    # 
    offset_a, offset_b = 81, 93
    # 
    # Gets MDNBR, heat flux, axial location, rod, and subchannel
    # from a row.
    # 
    get_values = itemgetter([-3, -4, -5, -2, -1])
    
    with open('test.csv', 'w') as result_file:
        writer = csv.writer(result_file, delimiter=';')
        for path in glob(os.path.join(folder_path, '*.out')):
            with open(path, 'r') as lines:
                chf_lines = dropwhile(
                    lambda line: not line.startswith('Summary of CHF Data'),
                    lines
                )
                writer.writerows(
                    [
                        get_values(islice(chf_lines, n, n + 1).next().split())
                        for n in [offset_a, offset_b - offset_a]
                    ]
                )


if __name__ == '__main__':
    main()