Discussion :

[manaliac]

Bonjour tout le monde,

Je cherche à lire un fichier scientifique avec deux types d'informations répétitives pour chaque element i:

element=element_i
1,2,3
5,9,8
....
....
load_element_i, definition1
definition2
defintion3

Une fois arrivée au mot load_element_i, je voudrais supprimer la ligne ou il se trouve et les 2 autres lignes qui suivent.

Je n'arrive pas à formuler une condition suffisante pour réussir mon but, et je n'ai plus touché le python depuis un bail!

Si vous avez une idée de comment faire, je vous en serai reconnaissante!

Merci.

M.

[VinsS]

Salut,

Tu peux le traiter ligne par ligne.

Exemple:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
 
outlines = []
count = 0
with open('myfile', 'r') as objf:
    lines = objf.readlines()
    while count < len(lines) - 1:
        if lines[count].startswith('load_element_'):
            count += 3
        else:
            outlines.append(lines[count])
            count += 1
 
with open('myfile', 'w') as objf:
    for line in outlines:
        objf.write(line)

Pas testé.

[mont29]

Il y a même moyen de faire ça sans lire tout le fichier d’un coup (important s’il est imposant) :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
with open('infile', 'r') as inf, open('outfile', 'w') as outf:
    decr = 0
    for l in inf:
        if decr:
            decr -= 1
        elif l.startswith("load_element_"):
            decr = 2
        else:
            outf.write(l + '\n')

code py3, pas testé non plus (mais je doute qu’il pose problème :p ).

À noter, ce code écrit dans un nouveau fichier, si on veut écraser le contenu du même fichier, soit on fait comme proposé par VinS (stockage des lignes dans une liste temporaire –*mais on retrouve le même problème de mémoire avec les gros fichiers)*:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
outl = []
with open('myfile', 'r') as inf:
    decr = 0
    for l in inf:
        if decr:
            decr -= 1
        elif l.startswith("load_element_"):
            decr = 2
        else:
            outl.append(l)
 
with open('myfile', 'w') as outf:
    outf.write('\n'.join(outl))

… Soit on utilise un fichier de sortie temporaire, puis on efface l’original et renomme le temporaire (plus adapté avec de gros fichiers)*:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
with open('infile', 'r') as inf, open('outfile', 'w') as outf:
    decr = 0
    for l in inf:
        if decr:
            decr -= 1
        elif l.startswith("load_element_"):
            decr = 2
        else:
            outf.write(l + '\n')
 
import os
os.remove("infile")
os.rename("outfile", "infile")