根据索引数组从DataFrame获取行

Question

我有一个数字数组，对应于需要从DataFrame中选择的行号。 例如，arr = np.array([0,0,1,1])和DataFrame如下所示。 arr是行号而不是索引。

Index   A   B   C   D
3      10   0   0   0
4      5    2   0   0

使用arr我想生成一个看起来像这样的DataFrame

  Index    A    B   C   D
    3      10   0   0   0
    3      10   0   0   0
    4      5    2   0   0
    4      5    2   0   0

Answer 1

您可以将iloc与整数索引一起使用：

df.iloc[[0,0,1,1], :]   # or df.iloc[arr, :]

#    A  B   C   D
#Index              
#3  10  0   0   0
#3  10  0   0   0
#4   5  2   0   0
#4   5  2   0   0