使用apply_along_axis的解决方案
可以通过使用^{}来实现解决方案:import numpy as np
def grayscale(colors):
"""Return grayscale of given color."""
r, g, b = colors
return 0.07 * r + 0.72 * g + 0.21 * b
image = np.random.uniform(255, size=(10,10,3))
result = np.apply_along_axis(grayscale, 2, image)
实例
10x10图像
我们现在可以开始可视化结果:from matplotlib import pyplot as plt
plt.subplot(1,2,1)
plt.imshow(image)
plt.subplot(1,2,2)
plt.imshow(result, cmap=gray)
文本示例(2x2图像)
要在文本中显示实际结果,我将使用一个较小的数组,仅使用一个2x2图像:image = np.random.uniform(250, size=(2,2,3))
内容是:array([[[205.02229826, 109.56089703, 163.74868594],
[ 11.13557763, 160.98463727, 195.0294515 ]],
[[218.15273335,