weixin_50751578的博客

原创 numpy整数数组索引

np.arange(3)[:,None]是一个[3,1]的矩阵，np.arange(2)[None,:]是一个[1,2]的矩阵。上面两个矩阵分别表示x第一个维度、第二个维度的索引值，最后得到[2,3]的索引结果。由于第三个维度为全切片，所以最终y的索引数组大小为[2,3,3]。索引矩阵在原数组每个维度的大小应该相同。上例中x有两个维度，y的两个索引大小都为[3,]，一个存储横坐标，一个存储纵坐标。总结：numpy会先检查每个维度的索引数组大小，如果大小不相等就先广播复制为相等大小的矩阵再做切片。