Суммировать столбцы с одинаковой строкой в ​​имени столбца ⇐ Python

[Anonymous]

Я хочу суммировать (по горизонтали) некоторые столбцы, имена которых имеют общую строку. Например, учитывая следующие столбцы:
a_red, a_blue, a_green, b_red, b_yellow, b_blue

Я хотел бы добавить в набор данных следующие столбцы:
a_tot = a_red + a_blue + a_green
b_tot = b_red + b_yellow + b_blue

Я попробовал использовать следующий код, но он выдает ошибку:
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


Подробнее здесь: https://stackoverflow.com/questions/798 ... olumn-name