新结果
简短的结果是
np.choose(labelling,im.transpose(2,0,1))
旧结果
尝试这个
im[np.arange(288)[:,None],np.arange(384)[None,:],labelling]
它适用于以下情况:
import numpy
import numpy.random
import itertools
a = numpy.random.randint(5,size=(2,3,4))
array([[[4, 4, 0, 0],
[0, 4, 1, 1],
[3, 4, 4, 2]],
[[4, 0, 0, 2],
[1, 4, 2, 2],
[4, 2, 4, 4]]])
b = numpy.random.randint(4,size=(2,3))
array([[1, 1, 0],
[1, 2, 2]])
res = a[np.arange(2)[:,None],np.arange(3)[None,:],b]
array([[4, 4, 3],
[0, 2, 4]])
# note that zip is not doing what you expect it to do
result = np.zeros((2,3))
for x,y in itertools.product(range(2),range(3)):
result[x,y] = a[x,y,b[x,y]]
array([[4., 4., 3.],
[0., 2., 4.]])
请注意,这zip
没有按照您的预期进行
zip(range(2),range(3))
[(0, 0), (1, 1)]
可能你的意思是itertools.product
list(itertools.product(range(2),range(3)))
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]
可怕的外观[:,None]
等可以通过使用来避免numpy.ix_
xx,yy = np.ix_( np.arange(2), np.arange(3) )
res = a[xx,yy,b]