首先,我们需要创建一个数据框,其中提供了一些数字值。接下来,我们需要使用apply函数,以每一行数据作为参数,传递一个lambda函数,该函数使用argmax函数来返回具有最大值的列索引。最后,我们可以使用index函数来获取对应的列名。 下面是代码示例:
import pandas as pd
# 创建一个数据框
df = pd.DataFrame({'col1': [1, 2, 3, 4],
'col2': [5, 6, 7, 8],
'col3': [9, 10, 11, 12]})
# 使用apply函数并传递一个lambda函数,以每一行数据作为参数
max_cols = df.apply(lambda x: x.argmax(), axis=1)
# 使用index函数获取对应的列名
max_cols_names = df.columns[max_cols]
print(max_cols_names)
输出:
Index(['col3', 'col3', 'col3', 'col3'], dtype='object')
上一篇:按行或按日期开发功能
下一篇:按行id分组的值的计数