Pandas update multiple columns using apply function
You can return pd.Series
:
dp_data_df = pd.DataFrame({'A':[3,5,6,6],
'B':[6,7,8,9],
'P':[5,6,7,0],
'Y':[1,2,3,4]})
print (dp_data_df)
A B P Y
0 3 6 5 1
1 5 7 6 2
2 6 8 7 3
3 6 9 0 4
def update_row(row):
listy = [1,2,3]
return pd.Series(listy)
dp_data_df[['A', 'P','Y']] = dp_data_df.apply(update_row, axis=1)
print (dp_data_df)
A B P Y
0 1 6 2 3
1 1 7 2 3
2 1 8 2 3
3 1 9 2 3
You can use zip your output:
def update_row(row):
listy = [1,2,3]
return listy
dp_data_df['A'], dp_data_df['P'], dp_data_df['Y'] = zip(*dp_data_df.apply(update_row, axis=1))