python, suma de valor de varias filas

Question

Tengo cientos de archivos csv como los siguientes:

Lo que quiero hacer es calcular la suma de cada fila para columnas A_*y B_*, respectivamente.

Por ejemplo file_1.csv, utilizo el siguiente código:

#for file_1.csv

import pandas as pd 
df = pd.read_csv('file_1.csv')
df["A_sum"]=df["A_1"]+df["A_2"]+df["A_3"]
df["B_sum"]=df["B_1"]+df["B_2"]
df.to_csv (r'output_file_1.csv', index = False, header=True)

Dado que tengo que procesar cientos de archivos como este, ¿cómo debo modificar el código para que pueda identificar automáticamente cuántas Acolumnas y bcolumnas en un archivo csv y calcular la suma?

Por ejemplo, puede generar código file_3.csvcomo este:

#for file_3.csv

import pandas as pd 
df = pd.read_csv('file_3.csv')
df["A_sum"]=df["A_1"]+df["A_2"]
df["B_sum"]=df["B_1"]+df["B_2"]
df.to_csv (r'output_file_3.csv', index = False, header=True)

Muchas gracias.

Answer 1

Puede utilizar filter:

ret_df = pd.DataFrame()

ret_df['A_sum'] = df.filter(like='A_').sum(1)
ret_df['B_sum'] = df.filter(like='B_').sum(1)

O use expresiones regulares y también un bucle:

for type in ['A','B']:
    df[f'{type}_sum'] = df.filter(regex=f'^{type}_').sum(1)