根据索引数组从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

- Bryce Ramgovind

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

- Psidom

---

不错。我认为在这种情况下 df.iloc[arr] 也可以使用。 - Ami Tavory