Python 获取具有最大值的列，这些列将考虑熊猫中的关系

让我们假设我们有了这个

我想知道哪一列每行的值最大

第1、2和3行的输出都是5列

第4行为

visions\u total

第5行是

['content\u gene\u strength'，'sport\u gene\u strength'，'visions\u total']

谢谢

按最大值比较所有列，然后使用带分隔符的列名称进行矩阵乘法，最后通过以下方式从右侧移除分隔符：

df['new'] = df.eq(df.max(axis=1), axis=0).dot(df.columns + ',').str.rstrip(',')