Pandas DataFrameで各行に対して関数を適用させる

Question

import pandas as pd

df = pd.DataFrame({
"A": [1,2,3,4],
"B": [5,6,7,8],
})

上記のようなDataFrameのデータがあるときに、各行に対して、下記のような関数を適用させて、その結果を、df["Result"] としてDataFrameに格納したいです。

def myFunc(val1: float, val2: float) -> float {
  # val1 と val2 を使った何らかの計算式とその結果を返す関数
  return xxx
}

for文を使えばできることは理解しているのですが、データが巨大になるとfor文だと遅いので、applyやapplyMapといったPandasの関数を利用できないかと思ったのですが、調べてもわかりませんでした。

どなたかお分かりになる方いたら教えていただけないでしょうか？

Answer 1

applyでaxis=1を指定するとできます。
参考：pandasで要素、行、列に関数を適用するmap, applymap, apply

import pandas as pd

def myFunc(val1: float, val2: float) -> float:
    xxx = val1 + val2
    return xxx

df = pd.DataFrame({
"A": [1,2,3,4],
"B": [5,6,7,8],
})

df["Result"] = df.apply(lambda row: myFunc(row[0], row[1]), axis=1)
print(df)
"""
   A  B  Result
0  1  5       6
1  2  6       8
2  3  7      10
3  4  8      12
"""