获取给定条件成立的数据帧的列号

时间:2017-12-21 20:57:02

标签: python pandas dataframe

我有以下代码:

<head>
  <meta charset="UTF-8">
  <title>
    <%= title %>|Minsk Events</title>
  <link rel="stylesheet" href="css/style.css">
</head>

打印:

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 个答案:

答案 0 :(得分:2)

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