重塑 matplotlib.pyplot.imread(*args, **kwargs) 返回的数组

Reshaping the array returned by matplotlib.pyplot.imread(*args, **kwargs)

matplotlib.pyplot.imread(*args, **kwargs) returns RGB 图像的 MxNx3(其中 M 和 N 是图像的高度和宽度,3 是 R、G 和 B 通道的数量)。我应该如何将输出重塑为 3xMxN?

假设arr作为输入数组,可以使用np.transpose-

arr.transpose(2,0,1)

np.rollaxis-

np.rollaxis(arr,2,start=0)