python - 如何在python中将两个不同维度的数组滚动为一维数组

Question

我有两个不同 mXn 维度的数组 (a,b)

我需要知道如何将这两个数组合并为一个一维数组

我对 a,b 数组都使用了 np.flatten()，然后将它们合并为一个数组，但我得到的是一个包含两个一维数组 (a,b) 的数组

a = np.array([[1,2,3,4],[3,4,5,6],[4,5,6,7]]) #3x4 array
b = np.array([ [1,2],[2,3],[3,4],[4,5],[5,6]]) #5x2 array

result = [a.flatten(),b.flatten()]
print(result)
[array([1, 2, 3, 4, 3, 4, 5, 6, 4, 5, 6, 7]), array([1, 2, 2, 3, ... 5, 6])]

在 matlab 中，我会这样做:

res = [a(:);b(:)]

另外，如何从结果中取回 a 和 b？

Answer 1

使用ravel + concatenate:

>>> np.concatenate((a.ravel(), b.ravel()))
array([1, 2, 3, 4, 3, 4, 5, 6, 4, 5, 6, 7, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6])

ravel 返回数组的一维 View ，是一种廉价操作。 concatenate 将 View 连接在一起，返回一个新数组。

---

顺便说一句，如果您希望能够取回这些数组，您需要将它们的形状存储在某个变量中。

i = a.shape
j = b.shape
res = np.concatenate((a.ravel(), b.ravel()))

稍后，要从 res 中检索 a 和 b，

a = res[:np.prod(i)].reshape(i)
b = res[np.prod(i):].reshape(j)

a
array([[1, 2, 3, 4],
       [3, 4, 5, 6],
       [4, 5, 6, 7]])

b 
array([[1, 2],
       [2, 3],
       [3, 4],
       [4, 5],
       [5, 6]])