我正在尝试弄清如何更改这样的数组:
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
a.shape
(3,3)
into:
a = np.array([[[1,255,255],[2,255,255],[3,255,255]],
[4,255,255],[5,255,255],[6,255,255]],
[7,255,255],[8,255,255],[9,255,255]]])
a.shape
(3,3,3)
基本上是将单个元素1
转换为[1, 255, 255]
我已经尝试使用reshape
,但似乎无法在没有慢速的for
循环的情况下完成此操作的逻辑。