Discussion :

[jahawai]

Bonjour à tous,

J'aimerais pouvoir créer et remplir des listes, en fonction des valeurs d'une colonne spécifique d'une liste initiale.
A titre d'exemple, la liste pourrait être :

Code :

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

list = [['Ingénieur','Pierre',30],['Chercheur','Etienne',40],['Chercheur','Sophie',25],['Ingénieur','Marie',54]]

Et je voudrais obtenir :

Code :

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

1
2
list1 = [['Ingénieur','Pierre',30],['Ingénieur','Marie',54]]
list2 = [['Chercheur','Etienne',40],['Chercheur','Sophie',25]]

Ma problématique est que je ne connais pas le contenu de la liste (contrairement à ce que je présente ici. Et elle peut très bien contenir également une ligne ['Technicien','Jean',42] ou tout autre élément.
Je connais seulement le nombre de colonnes de mon tableau.

J'aimerais donc pouvoir créer une nouvelle liste pour chaque élément différent de ma colonne 1, sans en savoir le nombre au départ.
Mais je ne sais pas s'il est possible (et comment) de créer ces listes "conditionnées".

Je précise que j'aimerais déjà essayer de faire cela sans numpy, si c'est jouable ;-)

Quelqu'un a t'il une idée pour moi ?

Merci d'avance !

[wiztricks]

Salut,

J'aimerais donc pouvoir créer une nouvelle liste pour chaque élément différent de ma colonne 1, sans en savoir le nombre au départ.
Mais je ne sais pas s'il est possible (et comment) de créer ces listes "conditionnées".

Avec un dictionnaire dont la clef serait la première colonne et la valeur associée la liste des éléments qui commencent avec cette chaine de caractères.

- W

[BufferBob]

salut,

les listes en intension (list comprehensions) permettent de facilement générer tes deux listes avec leurs discriminants :

Code :

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

1
2
3
>>> liste = [['Ingénieur','Pierre',30],['Chercheur','Etienne',40],['Chercheur','Sophie',25],['Ingénieur','Marie',54]]
>>> list1 = [i for i in liste if i[0] == 'Chercheur']
>>> list2 = [i for i in liste if i[0] == 'Ingénieur']

on peut aussi faire tout en une passe en s'appuyant sur itertools.groupby() et obtenir le dictionnaire qu'évoque wiztricks :

Code :

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

1
2
3
4
5
>>> from itertools import groupby
>>> liste = [['Ingénieur','Pierre',30],['Chercheur','Etienne',40],['Chercheur','Sophie',25],['Ingénieur','Marie',54]]
>>> dico = dict((i,list(j)) for i,j in  groupby(sorted(liste), key=lambda i:i[0]))
>>> dico['Ingénieur']
[['Ingénieur', 'Marie', 54], ['Ingénieur', 'Pierre', 30]]

[jahawai]

Bonjour à tous les deux,

Merci pour vos réponses, je vais regarder du côté des dictionnaires !

Ton premier exemple ne me convient pas BufferBob, car il requiert de connaître ce qu'il y a dans les listes (c'est comme ça que je faisais jusqu'à présent, mais maintenant, je veux également que le code soit capable de générer lui-même une nouvelle liste si un utilisateur lambda vient ajouter des nouvelles entrée).
Dans le 2ème exemple BufferBob, j'ai quand même l'impression qu'il faut connaître également la première colonne (au moins pour l'appeler derrière) mais je vais creuser ça.

En tous cas merci et je reviens vers vous éventuellement !

[jahawai]

Rebonjour,

Bon du coup, voilà le code qui me permet de faire ce que je voulais, en utilisant les dictionnaires et sans savoir, d'emblée, combien de valeurs différentes on aura dans la catégorie (ici "métier").
C'est probablement assez moche, je m'excuse d'avance pour vos yeux d'experts. Mais ça fonctionne...

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
16
17
18
from itertools import groupby
 
liste = [['Ingénieur','Pierre',30],['Chercheur','Etienne',40],['Chercheur','Sophie',25],['Ingénieur','Marie',54],['Chercheur','Sophie',24],['Technicien','Raymond',21]]
 
# On construit un dictionnaire qui prend pour clé la première valeur de chaque liste
dico = dict((i,list(j)) for i,j in  groupby(sorted(liste), key=lambda i:i[0]))
 
# On construit la liste des métiers 
liste_metiers = []
for i in range(0,len(liste)):
    liste_metiers.append(liste[i][0])
 
# A laquelle on enlève les doublons
liste_sans_doublons = list(set(liste_metiers))
 
# On renvoie une liste pour chaque métier
for i in range(0,len(liste_sans_doublons)):
    print(dico[liste_sans_doublons[i]])

Merci pour votre aide !

[fred1599]

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
16
17
18
19
20
job_list = [['Ingénieur','Pierre',30],
            ['Chercheur','Etienne',40],
            ['Chercheur','Sophie',25],
            ['Ingénieur','Marie',54],
            ['Chercheur','Sophie',24],
            ['Technicien','Raymond',21]]
 
 
 
job_dict = {}
for job in job_list:
    key, firstname, age = job
    values = (firstname, age)
    if key not in job_dict:
        job_dict[key] = [values,]
    else:
        job_dict[key].append(values)
 
 
print(job_dict) # {'Technicien': [('Raymond', 21)], 'Ingénieur': [('Pierre', 30), ('Marie', 54)], 'Chercheur': [('Etienne', 40), ('Sophie', 25), ('Sophie', 24)]}

Mais job_list peut être évité et on pourrait créer directement job_dict à la volée.