Pandas：如何聚合多個列？

Question

如何創建以兩列作為輸入並輸出分數的 function； 具體來說， 推土機的距離。

我找不到有關如何通過 groupby 實現它的文檔？

Answer 1

如果我了解您要完成的工作，您可以groupby.apply使用lambda或創建 function 和groupby.apply ：

from scipy.stats import wasserstein_distance
import pandas as pd
import numpy as np

df = pd.DataFrame([np.random.randint(0,10,2),
                   np.random.randint(0,10,2),
                   np.random.randint(0,10,2),
                   np.random.randint(0,10,2)], columns=list('AB'))

df['C'] = list('AABB')

# use apply with lambda function
print(df.groupby('C').apply(lambda df: wasserstein_distance(df['A'], df['B'])))