Fassen Sie Spalten mit derselben Zeichenfolge im Spaltennamen zusammen ⇐ Python

[Anonymous]

Ich möchte einige Spalten (horizontal) zusammenfassen, die eine gemeinsame Zeichenfolge in ihrem Namen haben. Zum Beispiel anhand der folgenden Spalten

Code: Select all

a_red, a_blue, a_green, b_red, b_yellow, b_blue

Ich möchte dem Datensatz die folgenden Spalten hinzufügen:

Code: Select all

a_tot = a_red + a_blue + a_green
b_tot = b_red + b_yellow + b_blue

Ich habe es mit dem folgenden Code versucht, aber er löst einen Fehler aus:

Code: Select all

df_loop = df
list_col_agg = ['a', 'b']
for i in list_col_agg:
df_1 = df_loop.with_column(pl.sum_horizontal(df_loop.select(
cs.contains(i)).columns).alias(f'tot_{i}'))
df_loop = df_1