np.transpose 改变维度索引值 比如下面这个例子,n是一个三维数组,第0维度与第1维度互换了 np.nonzero nonzero函数是numpy中用于得到数组array中非零元素的位置(数组索引)的函数 示例 n = numpy.array([4,3,2,1]).reshape((2,2,1)) p = numpy.array([0,0,2,1]).reshape((2,2,1)) n[numpy.nonzero( p)] –> array([2, 1])