将两个不同形状的numpy数组合并到一个数组中

Question

我有两个长度分别为 53 和 82 的 numpy 数组 a 和 b。我想将它们合并到一个数组中，因为我想使用 53+82=135 长度的数组，比如调用它 c 进行绘图。

我试过了

c = a+b 

但我收到 ValueError: shape mismatch: objects cannot be broadcast to a single shape

这可能吗？

Answer 1

你需要使用numpy.concatenate而不是数组加法

c = numpy.concatenate((a, b))

实施

import numpy as np
a = np.arange(53)
b = np.arange(82)
c = np.concatenate((a, b))

输出

c.shape
(135, )

Answer 2

使用numpy.concatenate:

In [5]: import numpy as np

In [6]: a = np.arange(5)                                                                         

In [7]: b = np.arange(11)                                                                        

In [8]: np.concatenate((a, b))                                                                   
Out[8]: array([ 0,  1,  2,  3,  4,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10])

对于一维数组，您也可以使用numpy.hstack：

In [9]: np.hstack((a, b))                                                                       
Out[9]: array([ 0,  1,  2,  3,  4,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10]