python - Dataframe 为一列选择 Max 但输出另一列的值

Question

我有一个数据框，其值类似于下面

A10d  B10d  C10d  A   B   C   Strategy
20    10    5     3   5   1    3

策略选择 A10d、B10d、C10d 中的最大值并返回 A、B、C 的值在这种情况下，A10d 是最大的，Strategy 返回 A，值为 3

我不确定如何正确创建此策略列，有人可以提供建议吗？非常感谢您的帮助

Answer 1

我想你需要iloc按位置选择第一列，然后使用 idxmax 按 max 获取列名称和 replace 10d 通过匹配列的空格。最后创建新列 lookup :

print (df)
   A10d  B10d  C10d  A  B  C
0    20    10     5  3  5  1
1    20   100     5  3  5  1

df1 = df.iloc[:,:3]
print (df1)
   A10d  B10d  C10d
0    20    10     5
1    20   100     5

s = df1.idxmax(axis=1).str.replace('10d','')
print (s)
0    A
1    B
dtype: object

df['Strategy'] = df.lookup(df.index, s)
print (df)
   A10d  B10d  C10d  A  B  C  Strategy
0    20    10     5  3  5  1         3
1    20   100     5  3  5  1         5