我有一个数字数组,它对应于需要从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
df.iloc[arr]
也可以使用。 - Ami Tavory