Créer n listes (où n est variable et dépend des valeurs d'une autre liste)

Version imprimable

26/03/2019, 11h16
jahawai

Créer n listes (où n est variable et dépend des valeurs d'une autre liste)
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:

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

Et je voudrais obtenir :
Code:

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 !
26/03/2019, 11h47
wiztricks

Salut,

Citation:

Envoyé par jahawai

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

salut,

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

Code:

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:

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]]

27/03/2019, 09h00
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 !

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:

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 !

Code:

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.

27/03/2019, 16h28
jahawai

Beaucoup plus propre merci !

Question subsidiaire : Comment faire si je veux utiliser non plus seulement le métier comme discriminant mais aussi la deuxième colonne (ici le prénom) ?
J'imagine que cela complique un peu les choses vu qu'on n'a droit qu'à une clé non ?

La solution que j'envisage est de mettre une liste en clé de dictionnaire... mais cela ne semble pas possible.
Je l'ai fait avec des tuples. Mais mon souci est que du coup, le premier élément de mes listes de résultats sont des tuples, ce qui me gêne un peu.
Je vais me débrouiller pour les rescinder en string, mais s'il y a plus simple...

Salut.

La clé d'un dict peut très bien être un tuple.

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
liste = [
    ['Ingénieur','Pierre',30],
    ['Chercheur','Etienne',40],
    ['Chercheur','Sophie',25],
    ['Ingénieur','Marie',54],
    ['Chercheur','Sophie',24],
    ['Technicien','Raymond',21]
]
 
dico = {}
for metier, prenom, *attrs in liste:
    dico.setdefault((metier, prenom), set()).add(tuple(attrs))
 
print(dico)

Ou alors faire un autre niveau de dict.

Code:

1
2
3
4
5
6
dico = {}
for metier, prenom, *attrs in liste:
    m = dico.setdefault(metier, dict())
    m.setdefault(prenom, set()).add(tuple(attrs))
 
print(dico)

27/03/2019, 16h40
fred1599

Je ne comprend pas l'intérêt conceptuel de ce que tu demandes.

Peut-être veux-tu récupérer l'âge en connaissant le métier et le prénom ?
27/03/2019, 17h01
jahawai

Non, non, je reconnais qu'il n'y a aucun intérêt conceptuel...
En fait, j'ai pris une liste complètement fictive pour expliquer mon problème mais cela n'a pas vraiment de sens, je le conçois.

Pour expliquer plus globalement mon problème, voici quelques éléments. Désolé, je ne peux pas communiquer mes données pour que ce soit vraiment concret.
J'ai un tableau de données (un csv pour être précis) dont les trois premières colonnes sont des éléments qui permettent de classifier un individu et les colonnes suivantes sont des informations complémentaires sur lesquelles je vais chercher à faire des calculs par la suite. Ce tableau peut évoluer car par la suite, un utilisateur pourra être amené à agrandir la base de données.
Je veux pouvoir appliquer un code de calcul (écrit par ailleurs) à un individu (qualifié par les trois première colonnes, ex : nom, prénom, id).
Pour faire cela, j'ai décidé de découper mon tableau (liste de listes) en autant de listes que d'individus présents dans le tableau au moment où on ouvre la bdd. Et j'appliquerai mon code de calcul à la liste correspondant à l'individu que je voudrai.

Je précise quand même que c'est un peu volontairement que je fais tout ça "à la bourrin" car j'essaye d'intégrer les bases de python en même temps.