J'ai cette fonction dans une classe

def readInput(self, inFile):
    with open(inFile, "r") as file:
        for line in file:
            if (line.split()[0] == "Ks"):
                nextLine = next(file)
                self.Ks = nextLine.split()[0]
                self.thetaS = nextLine.split()[1]
                self.thetaR = nextLine.split()[2]
                self.alpha = nextLine.split()[3]
                self.lamb = nextLine.split()[4]
                self.n = nextLine.split()[5]

Il recherche essentiellement un modèle ("Ks") à l'intérieur d'un fichier d'entrée (inFile) pour stocker des valeurs de variable à partir de la ligne suivante de inFile à l'intérieur des variables d'instance.

Il y a beaucoup de répétitions dans le code et je pense qu'il est possible d'écrire cela de manière intelligente (et plus courte).

Des idées?

Le fichier d'entrée ressemble à ceci:

### Soil hydraulic properties. Units: m, d
Ks      ThetaS  ThetaR  alpha   lambda  n
0.128   0.01    0.42    0.84    -1.497  1.441
0
Andre Herman Bezerra 15 mars 2019 à 20:58

2 réponses

Meilleure réponse

Utilisation du déballage de tuple:

self.Ks, self.thetaS, self.thetaR, self.alpha, self.lamb, self.n = nextLine.split()
5
Alex Hall 15 mars 2019 à 17:59

Une autre possibilité consiste à utiliser le package CSV standard.

import csv

with open('testdata.csv', newline='') as file:
    csvreader = csv.reader(file, delimiter=' ', skipinitialspace=True)
    next(csvreader)  # omit the comment at start of file
    header = next(csvreader)
    for row in csvreader:
        print(', '.join(row))
2
handras 15 mars 2019 à 20:31