Python 获取给定条件所在的dataframe的列号
我有以下代码:Python 获取给定条件所在的dataframe的列号,python,pandas,dataframe,Python,Pandas,Dataframe,我有以下代码: 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 与此相反,我只想得到一个索引列表,其中包含此条件,如下所示
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]
知道我该怎么做吗?回答得好。。。太完美了+1.
wrong_indexes = df.loc[df['cA'] > 2 ].index.tolist()