Discussion :

[beefjoy]

Bonjour à tous,
Voila mon pb, j'ai un tableau de ce type:

***** Axes restored to original set *****
-------------------------------------------------------------------
Center Atomic Forces (Hartrees/Bohr)
Number Number X Y Z
-------------------------------------------------------------------
1 6 0.000000000 0.000000000 0.032384058
2 1 0.000000000 0.008075832 -0.016192029
3 1 0.000000000 -0.008075832 -0.016192029
-------------------------------------------------------------------

et je voudrais obtenir de celui-ci les colonnes suivantes:

6 0.000000000 0.000000000 0.032384058
1 0.000000000 0.008075832 -0.016192029
1 0.000000000 -0.008075832 -0.016192029

je sais à peu près faire avec un script awk, à base de printf $2 $3 etc. et je souhaiterais être capable de faire la même chose en python. Je ne trouve pas de méthodes adaptées ou même la bonne approche.

Je vous serais très reconnaissant pour votre aide.
Alban.

[__dardanos__]

Salut.
Voici de quoi te dépanner :

Code :

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
def extraire_donnees(filename):
    tab = []
    for line in open(filename,'r'):
        try:
            v = map(float,line.split())
        except (ValueError):
            pass
        else:
            tab.append(v[1:])
    return tab
 
if __name__ == '__main__':
    for v in extraire_donnees('fichier.in'):
        print v

[beefjoy]

merci __dardanos__ pour ta réponse !
je vais de ce pas essayer ta function.

[beefjoy]

ok, ça marche !
si un jour tu repasses par ce post, j'ai une question:

j'ai commencé python avec la version 3, et quand j’exécute ton script j'obtiens:
TypeError: 'map' object is not subscriptable
j'ai vu que la fonction map avait changé de la version 2 à 3 (d'ailleurs ta fonction marche parfaitement en python 2) et qu'il fallait convertir l' "iterator object" retourné par map en liste avant de pouvoir l'imprimer. Mais c'est ce que je croyais que tu avais fait ....

saurais-tu m'expliquer quelle est la différence et comment convertir le script pour qu'il fonctionne en python 3 ?

merci d'avance.

[beefjoy]

ok, apparemment:

v = list(map(float, line.split()))

règle le pb en python 3.

[__dardanos__]

Salut,
Avec Python 3, map retourne un iterator. Il faut par exemple le convertir en liste comme tu l'as fait, ou alors utiliser une comprehension list :

Code :

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
def extraire_donnees(filename):
    tab = []
    for line in open(filename,'r'):
        try:
            v = [ float(e) for e in line.split() ]
        except (ValueError):
            pass
        else:
            tab.append(v[1:])
    return tab
 
if __name__ == '__main__':
    for v in extraire_donnees('fichier.in'):
        print(v)