如何创建取两列均值与其中一列最大值的新forecast列？

你可以用以下两种方法实现需求：

方法一：使用numpy的maximum函数（推荐，效率更高） #

利用np.maximum逐行对比A、B列的均值与A列的值，取较大者作为新列值：

import pandas as pd
import numpy as np

# 若已有df3可跳过示例数据生成
df3 = pd.DataFrame({'A': [10, 20, 30], 'B': [5, 25, 15]})

# 创建forecast列
df3['forecast'] = np.maximum(df3[['A', 'B']].mean(axis=1), df3['A'])

执行后df3会新增forecast列，每一行的值是对应行A、B均值和A值中的较大者。

方法二：使用apply方法（适合小数据集） #

通过apply逐行计算最大值：

df3['forecast'] = df3.apply(lambda row: max(row[['A', 'B']].mean(), row['A']), axis=1)

注意：大数据量下优先选方法一，apply的循环逻辑效率较低。

内容的提问来源于stack exchange，提问作者spcol