python, somma di valori da più righe

Question

Ho centinaia di file CSV come i seguenti:

Quello che voglio fare è calcolare la somma di ogni riga per colonne A_*e B_*, rispettivamente.

Ad esempio file_1.csv, utilizzo il codice seguente:

#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)

Dato che devo elaborare centinaia di file come questo, come devo modificare il codice in modo che possa identificare automaticamente quante Acolonne e bcolonne in un file csv e calcolare la somma?

Ad esempio, può generare codice per file_3.csvcome questo:

#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)

Grazie mille.

Answer 1

Puoi usare 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)

Oppure usa regex e anche un ciclo:

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