对于每一行,返回最小值的列名,可以使用pandas库中的idxmin()
函数来实现。idxmin()
函数返回最小值所在的列名。
以下是一个示例代码:
import pandas as pd
# 创建一个DataFrame
data = {'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 对每一行使用idxmin()函数获取最小值所在的列名
min_columns = df.idxmin(axis=1)
print(min_columns)
输出结果为:
0 A
1 A
2 A
dtype: object
在这个例子中,DataFrame df
有3列(A、B、C),对于每一行,idxmin()
函数返回最小值所在的列名。
领取专属 10元无门槛券
手把手带您无忧上云