获取满足条件的Pandas DataFrame行的整数索引？

Question

我有以下DataFrame：

   a  b  c
b
2  1  2  3
5  4  5  6

如您所见，列b用作索引。我想获取该行的序数('b' == 5)，在这种情况下应该是1。

被测试的列可以是索引列（b在这种情况下也可以）或常规列，例如，我可能想找到满足条件的行的索引('c' == 6)。

Answer 1

您可以这样使用np.where：

import pandas as pd
import numpy as np
df = pd.DataFrame(np.arange(1,7).reshape(2,3),
                  columns = list('abc'), 
                  index=pd.Series([2,5], name='b'))
print(df)
#    a  b  c
# b         
# 2  1  2  3
# 5  4  5  6
print(np.where(df.index==5)[0])
# [1]
print(np.where(df['c']==6)[0])
# [1]

返回的值是一个数组，因为一列中可能有多个具有特定索引或值的行。