当DataFrame包含字符串时，是否存在与DataFrame.idxmax等效的值？

Question

我正在使用pandas读取选举结果的简单CSV文件：

constituency,anug,apnuafc,cg,ljp,pppc,...
Barima-Waini,0,3905,0,170,8022,...
Pomeroon-Supenaam,86,7343,149,120,18788,...
Essequibo Islands-West Demerara,310,23811,318,0,47855,...
...

我使用views.py中的election.votes访问此文件：

results = pd.read_csv(election.votes)

对于每一行，我想为获胜方添加一个新列。我试过：

results["winner"] = results.max(axis=1)

但这会添加最高的值，而不是相应的列标题。所以我试过：

results["winner"] = results.idxmax(axis=1)

然后我得到错误reduction operation 'argmax' not allowed for this dtype

由于选区的限制，我不能用to_numeric使idxmax发挥作用

是否有其他有效的方法获取列标题

Answer 1

对仅获取数值列使用^{}：

import numpy as np

results["winner"] = results.select_dtypes(np.number).idxmax(axis=1)