我有一个数组A
有形状的(480, 640, 3)
,和一个数组B
有形状(480, 640)
.
如何将这两个附加为一个具有形状的数组(480, 640, 4)
?
I tried np.append(A,B)
但它不保留维度,而axis
选项导致ValueError: all the input arrays must have same number of dimensions
.
Use dstack http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.dstack.html:
>>> np.dstack((A, B)).shape
(480, 640, 4)
这可以处理数组具有不同维数并沿第三轴堆叠数组的情况。
否则,要使用append
or concatenate
,你必须做B
自己绘制三维并指定要连接它们的轴:
>>> np.append(A, np.atleast_3d(B), axis=2).shape
(480, 640, 4)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)