比如现在我有 a = np.array([[1,2,3],[4,5,6],[7,8,9]])
然后我有另一个数组 b = np.array([0,2,0]) 即为每行中我想取的那一列
如何快速得到c = [1,8,3]?
或者一个W x H x C的RGB图像I,现在我有一个W x H的矩阵m,里面的值为0,1或2,即为 个行列中我想取的那个通道
如何和快速得到相应的W x H的矩阵f?
除了写循环遍历以外有没有别的方法?
比如现在我有 a = np.array([[1,2,3],[4,5,6],[7,8,9]])
然后我有另一个数组 b = np.array([0,2,0]) 即为每行中我想取的那一列
如何快速得到c = [1,8,3]?
或者一个W x H x C的RGB图像I,现在我有一个W x H的矩阵m,里面的值为0,1或2,即为 个行列中我想取的那个通道
如何和快速得到相应的W x H的矩阵f?
除了写循环遍历以外有没有别的方法?