相关疑难解决方法(0)

我想将具有形状的灰度图像转换为具有形状(height,width)的3通道图像(height,width,nchannels).工作是用a完成的for-loop,但必须有一个简洁的方法.这是程序中的一个代码,有人可以给出一个提示.请指教.

 30         if img.shape == (height,width): # if img is grayscale, expand
 31             print "convert 1-channel image to ", nchannels, " image."
 32             new_img = np.zeros((height,width,nchannels))
 33             for ch in range(nchannels):
 34                 for xx in range(height):
 35                     for yy in range(width):
 36                         new_img[xx,yy,ch] = img[xx,yy]
 37             img = new_img