在Python中为不同的ndarray索引分配相同的值

Question

问题可能不清楚，所以我你用代码解释：

@staticmethod
def generate_data_uncertainty_heat_map(data_property, data_uncertainty, file_path):
    plt.figure()
    uncertainty = numpy.zeros((data_property.rows, data_property.cols, 4), 'uint8')
    uncertainty[..., 0] = uncertainty[..., 1] = uncertainty[..., 2] = numpy.uint8(data_property * 255)
    uncertainty[..., 3] = numpy.uint8(data_uncertainty)
    fig = plt.imshow(uncertainty, extent=[0, data_property.cols, data_property.rows, 0])
    plt.colorbar(fig)
    plt.savefig(file_path + '.png')
    plt.close()

这需要两个（n，n）ndarray来形成一个带有matploblib的RGBA图像。 为此，我使用data_属性参数作为我的RGB，数据不确定性作为我的不透明度。在

我基本上想知道我是否会写作

^{pr2}$

另一方面，对于不确定度[…，0或1或2]来说，该值应该是纽比.uint8（数据属性*255）。在

提前谢谢你。在

Answer 1

您可以使用data_property[..., np.newaxis]在右侧添加一个新轴。然后任务可以这样完成：

uncertainty[..., :3] = numpy.uint8(data_property * 255)[..., np.newaxis]

---

例如

^{pr2}$