Python: 如何對 pandas.DataFrame 兩欄位運算後,增加到最後一欄? df[‘sum_AB’] = df.apply(sum_ab, axis=1) ; lambda函式

import pandas as pd
import numpy as np

# 創建一個包含三個欄位的 DataFrame
df = pd.DataFrame({‘A’: [1, 2, 3], ‘B’: [4, 5, 6], ‘C’: [7, 8, 9]})

# 自定義函數計算 A 和 B 欄位的總和，然後將結果新增到 DataFrame 中
def sum_ab(row):
return row[‘A’] + row[‘B’]

df[‘sum_AB’] = df.apply(sum_ab, axis=1)
print(df)

.apply()中

可使用lambda函式

df[‘sum_AB’] = df.apply(lambda row: row[‘A’] + row[‘B’], axis=1)

推薦hahow線上學習python: https://igrape.net/30afN

df_Frame_data1:

df_Frame_data1[“r”] = df_Frame_data1.apply(lambda df:

(df[“X”]**0.5 + df[“Y”]**0.5)**0.5, axis=1) #X**2+Y**2的筆誤

預設值為axis=0,

若沒寫axis=1 , 會出現 KeyError: ‘X’ 

如果不熟悉apply()

就要取出X, Y

經過運算後生成新的ｒ

再將 [原DF , r] 

concat( axis=1 ) 起來

推薦hahow線上學習python: https://igrape.net/30afN