在numpy中将3个阵列组合在一个3D阵列中-Combining3arraysinone3Darrayinnumpy

9  

You can use numpy.dstack:

你可以使用numpy.dstack：

>>> import numpy as np
>>> a = np.random.random((11, 13))
>>> b = np.random.random((11, 13))
>>> c = np.random.random((11, 13))
>>> 
>>> d = np.dstack([a,b,c])
>>> 
>>> d.shape
(11, 13, 3)
>>> 
>>> a[1,5], b[1,5], c[1,5]
(0.92522736614222956, 0.64294050918477097, 0.28230222357027068)
>>> d[1,5]
array([ 0.92522737,  0.64294051,  0.28230222])

5  

numpy.dstack stack the array along the third axis, so, if you stack 3 arrays (a, b, c) of shape (N,M), you'll end up with an array of shape (N,M,3).

numpy.dstack沿第三轴堆叠数组，因此，如果堆叠3个形状（a，b，c）的形状（N，M），你最终会得到一个形状数组（N，M，3） 。

An alternative is to use directly:

另一种方法是直接使用：

np.array([a, b, c])

That gives you a (3,N,M) array.

这给你一个（3，N，M）数组。

What's the difference between the two? The memory layout. You'll access your first array a as

两者有什么区别？内存布局。你将访问你的第一个数组a

np.dstack([a,b,c])[...,0]

or

要么

np.array([a,b,c])[0]