python - 我们如何连接两个吐出的数组

Question

假设我们有数组 x

x = np.array([1, 2, 3, 99, 99, 3, 2, 1])
x1, x2, x3 = np.split(x, [3, 6])

我想使用附加这两个拆分的数组让我们说 x1 和 x2 我想像 [1, 2, 3, 2, 1] 一样连接如果我用于连接

x4 = x1+x3
x4

我们收到类似 ValueError 的错误：操作数无法与形状一起广播 (3,) (2,)

我们可以为连接两个拆分数组做些什么。你能帮我解决这个问题吗

标签： pythonarrayssplit

Answer 1

您可以使用np.hstack来获得所需的结果：

import numpy as np
x = np.array([1, 2, 3, 99, 99, 3, 2, 1])
x1, x2, x3 = np.split(x, [3, 6])
x4 = np.hstack((x1, x3))
print(x4)  # [1 2 3 2 1]

hstack代表水平堆栈，也有vstack垂直堆栈，这些函数也可以用于多维数组。