重塑matplotlib.pyplot.imread返回的数组（* args，** kwargs）

Question

matplotlib.pyplot.imread(*args, **kwargs)为RGB图像返回MxNx3（其中M和N是图像的高度和宽度，3是R，G和B通道的数量）。我应该如何将输出重新整形为3xMxN？

Answer 1

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

arr.transpose(2,0,1)

或np.rollaxis -

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