Numpy多维数组切片

Question

我有一个形状为(100,50,20)的3d NumPy数组。我试图通过使用索引来切片数组的第三维，例如，从1到6和从8到10。

我尝试了下面的代码，但它一直报告语法错误。

newarr [:,:,1:10] = oldarr[:,:,[1:7,8:11]]

Answer 1

您可以使用np.r_来连接切片对象:

newarr [:,:,1:10] = oldarr[:,:,np.r_[1:7,8:11]] 

---

示例:

np.r_[1:4,6:8]
array([1, 2, 3, 6, 7])