私は次のコードを持っている与えられた条件が

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()