使用来自其他列的两个计算值的最大值创建 Pandas 列

Question

我想创建一个列，其最大值介于从数据框的其他列计算的 2 个值之间。

import pandas as pd
df = pd.DataFrame({"A": [1,2,3], "B": [-2, 8, 1]})

df['Max Col'] = max(df['A']*3, df['B']+df['A'])


ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

期望的结果是具有上述计算最大值的新 df 列 ['Max Col']。

我知道用计算创建两个新列然后应用.max(axis=1)的长期解决方案。 我正在寻找一个直接的解决方案。

谢谢。

Answer 1

使用np.maximum ：

df['max'] =np.maximum(df['A']*3, df['B']+df['A'])

Output：

   A  B  max
0  1 -2    3
1  2  8   10
2  3  1    9

Answer 2

import pandas as pd
df = pd.DataFrame({"A": [1,2,3], "B": [-2, 8, 1]})
# map the max function to a zip of your calculations
df['max'] = list(map(max, zip(df['A']*3, df['B']+df['A'])))
print(df)

   A  B  max
0  1 -2    3
1  2  8   10
2  3  1    9

Answer 3

您可以使用apply方法 -

df['max'] = df.apply(lambda x: max(x['A']*3, x['A'] + x['B']), axis=1)