Question

我想找到包含特定值的列的索引。我有大数据集（超过2500列），值可以位于任何列中。

因此，我不知道该值所在的列的名称。数据集很大，超过2500列，并且值可以在2500列中的任何一个。

我需要列索引，而不是行索引

这只是一个示例数据集：

import pandas as pd

#THIS IS THE EXAMPLE ONLY, DATASET IS VERY BIG 
df = pd.DataFrame({'one':[1,2,3,4,5],
                   'two':[45,4,3,2,4],
                   'three':[4,2,121,111,5]})
print(df)

i = df[df==111].index
print(i)
#RESULT SHOULD BE INDEX 2

Answer 1

尝试stack和idxmax

n = (df.stack() == 111).idxmax()[1]
df.columns.get_loc(n)

Out[572]: 2

Answer 2

这应该有效：

df.eq(111).idxmax().max()

获取包含特定值的列的索引熊猫

2 个答案: