Repérer identifiants et catégories avec nunique

Partie 5 · ★★★

.nunique() compte les valeurs distinctes. Si une colonne en a autant que de lignes, c'est un identifiant ; si elle en a très peu, c'est une catégorie. Calcule nunique (dict colonne → nombre de valeurs distinctes) pour toutes les colonnes, puis la liste identifiants des colonnes qui ont autant de valeurs distinctes que de lignes.

employes = [
    {"id": 1, "nom": "Alice", "age": 25, "ville": "Paris", "salaire": 42000},
    {"id": 2, "nom": "Bob", "age": 31, "ville": "Lyon", "salaire": 38000},
    {"id": 3, "nom": "Chloé", "age": 45, "ville": "Toulouse", "salaire": 52000},
    {"id": 4, "nom": "David", "age": 35, "ville": "Paris", "salaire": 61000},
    {"id": 5, "nom": "Eve", "age": 22, "ville": "Lille", "salaire": 33000},
]
colonnes = list(employes[0])
nunique = {c: len(___(e[c] for e in employes)) for c in colonnes}
identifiants = [c for c in colonnes if nunique[c] == ___(employes)]

Ctrl/⌘+Entrée