如何在 Python 中连接几个（超过 2 个）2D arrays？

Question

我有以下 2 组 arrays 我只想按以下顺序连接它们：

在第一种情况下，我只想获取 a 的前两个元素和 b 的所有元素。

输入

a = [[1,1],[1,1],[1,1]]
b = [[2,2],[2,2]]

预期 OUTPUT

c = [[1,1],[1,1],[2,2],[2,2]]

还有第二种情况

输入

a = [1,1]
b = [2,2]

预期 OUTPUT

c = [1,1,2,2]

我也接受 numpy 解决方案。

Answer 1

您可以只使用+运算符

>>> a = [[1,1],[1,1]]
>>> b = [[2,2],[2,2]]
>>> a + b
[[1, 1], [1, 1], [2, 2], [2, 2]]

>>> a = [1,1]
>>> b = [2,2]
>>> a + b
[1, 1, 2, 2]

由于您也提到了 numpy，您可以使用numpy.concatenate

>>> np.concatenate([a, b])