获取给定条件所在的dataframe的列号

Question

我有以下代码：

raw_data = [[1, 2, 3], [4, 5, 6]]
df = pd.DataFrame(data=raw_data,
                  columns=["cA", "cB", "cC"])

wrong_indexes = df.loc[df['cA'] > 2 ]
print(wrong_indexes)

这张照片是：

   cA  cB  cC
1   4   5   6

与此相反，我只想得到一个索引列表，其中包含此条件，如下所示：

[1]

你知道我怎么做吗

Answer 1

wrong_indexes = df.loc[df['cA'] > 2 ].index.tolist()