Comment attribuer différentes valeurs de colonne à différentes variables en Python


chut

J'essaie d'affecter les trois groupes uniques de la groupcolonne dfà différentes variables (voir mon code) en utilisant Python. Comment incorporer cela dans une boucle for ? Évidemment var + ine fonctionne pas.

import pandas as pd

data = {
    'group': ['a', 'a', 'a', 'b', 'b', 'c', 'c'], 
    'num': list(range(7))
    }  
df = pd.DataFrame(data)  

unique_groups = df['group'].unique()

# How do I incorporate this logic inside a for loop??
var1 = df[df['group'] == unique_groups[0]]
var2 = df[df['group'] == unique_groups[1]]
var3 = df[df['group'] == unique_groups[2]]

# My approach:
for i in range(len(unique_groups)):
    var + i = df[df['group'] == unique_groups[i]] # obviously "var + i" does not work
Stuart

D'après votre commentaire, il semble qu'il soit acceptable all_varsd'être une liste, c'est-à- all_vars[0]dire le premier groupe, all_vars[1]le second, etc. Dans ce cas, envisagez d'utiliser à la groupbyplace :

all_vars = [group for name, group in df.groupby("group")]

Articles connexes