NumPy 沿不同轴将数组堆叠在一起

约定不等于承诺〃 2022-07-11 05:00 130阅读 0赞

### 组合(stack)不同的数组 ###

几种方法可以沿不同轴将数组堆叠在一起：

>>> a = floor(10*random.random((2,2)))
    >>> a
    array([[ 1., 1.], [ 5., 8.]])
    >>> b = floor(10*random.random((2,2)))
    >>> b
    array([[ 3., 3.], [ 6., 0.]])
    >>> vstack((a,b))
    array([[ 1., 1.], [ 5., 8.], [ 3., 3.], [ 6., 0.]])
    >>> hstack((a,b))
    array([[ 1., 1., 3., 3.], [ 5., 8., 6., 0.]])

函数 `column_stack` 以列将一维数组合成二维数组，它等同与 `vstack` 对一维数组。

>>> column_stack((a,b))   # With 2D arrays
    array([[ 1., 1., 3., 3.], [ 5., 8., 6., 0.]])
    >>> a=array([4.,2.])
    >>> b=array([2.,8.])
    >>> a[:,newaxis]  # This allows to have a 2D columns vector
    array([[ 4.], [ 2.]])
    >>> column_stack((a[:,newaxis],b[:,newaxis]))
    array([[ 4., 2.], [ 2., 8.]])
    >>> vstack((a[:,newaxis],b[:,newaxis])) # The behavior of vstack is different
    array([[ 4.], [ 2.], [ 2.], [ 8.]])

`row_stack` 函数，另一方面，将一维数组以行组合成二维数组。

对那些维度比二维更高的数组， `hstack` 沿着第二个轴组合， `vstack` 沿着第一个轴组合, `concatenate` 允许可选参数给出组合时沿着的轴。

Note

在复杂情况下， `r_[]` 和 `c_[]` 对创建沿着一个方向组合的数很有用，它们允许范围符号(“:”):

>>> r_[1:4,0,4]
    array([1, 2, 3, 0, 4])

当使用数组作为参数时， `r_` 和 `c_` 的默认行为和 `vstack` 和 `hstack` 很像，但是允许可选的参数给出组合所沿着的轴的代号。

更多函数hstack , vstack, column\_stack , row\_stack , concatenate , c\_ , r\_ 参见 [ NumPy示例][NumPy] .

0  
>>> a = floor(10\*random.random((2,2)))  
>>> a  
array(\[\[ 0.,  5.\],  
       \[ 1.,  2.\]\])  
>>> b = floor(10\*random.random((2,2)))  
>>> b  
array(\[\[ 1.,  6.\],  
       \[ 5.,  5.\]\])  
>>> vstack((a,b))  
array(\[\[ 0.,  5.\],  
       \[ 1.,  2.\],  
       \[ 1.,  6.\],  
       \[ 5.,  5.\]\])  
>>> hstack((a,b))  
array(\[\[ 0.,  5.,  1.,  6.\],  
       \[ 1.,  2.,  5.,  5.\]\])  
>>> column\_stack((a,b))  
array(\[\[ 0.,  5.,  1.,  6.\],  
       \[ 1.,  2.,  5.,  5.\]\])  
>>> a=array(\[4.,2.\])  
>>> b=array(\[2.,8.\])  
>>> a\[:,newaxis\]  
array(\[\[ 4.\],  
       \[ 2.\]\])  
>>> a  
array(\[ 4.,  2.\])  
>>> b  
array(\[ 2.,  8.\])  
>>> b\[:,newaxis\]  
array(\[\[ 2.\],  
       \[ 8.\]\])  
>>> column\_stack((a\[:,newaxis\],b\[:,newaxis\]))  
array(\[\[ 4.,  2.\],  
       \[ 2.,  8.\]\])  
>>> a=array(\[4.,3.\])  
>>> column\_stack((a\[:,newaxis\],b\[:,newaxis\]))  
array(\[\[ 4.,  2.\],  
       \[ 3.,  8.\]\])  
>>> vstack((a\[:,newaxis\],b\[:,newaxis\]))  
array(\[\[ 4.\],  
       \[ 3.\],  
       \[ 2.\],  
       \[ 8.\]\])  
>>> row\_stack(a,b)  
Traceback (most recent call last):  
  File "<stdin>", line 1, in <module>  
TypeError: vstack() takes exactly 1 argument (2 given)  
>>> row\_stack((a,b))  
array(\[\[ 4.,  3.\],  
       \[ 2.,  8.\]\])  
>>> a  
array(\[ 4.,  3.\])  
>>> b  
array(\[ 2.,  8.\])  
>>> r\_\[1:4,0,4\]  
array(\[1, 2, 3, 0, 4\])  
>>>

[NumPy]: http://scipy.org/Numpy_Example_List