Python中的矩阵生成

Question

我想从 python 中的其他两个矩阵生成一个矩阵。所以我想做的是考虑第一个矩阵的第一行并将其放入新矩阵中，然后相同但来自第二个矩阵。下一步将移动到两个矩阵的第二行，依此类推。

输入：

A = [[1, 1, 1, 1, 1], [2, 2, 2, 2, 2], [3, 3, 3, 3, 3], [4, 4, 4, 4, 4], [5, 5 ,5 ,5, 5]]

B = [[6, 6, 6, 6, 6], [7, 7, 7 ,7 ,7], [8, 8, 8, 8, 8], [9, 9, 9, 9, 9], [0, 0, 0, 0, 0]]

应该产生输出：

C = [[1, 1, 1, 1, 1], [6, 6, 6, 6, 6,], [2, 2, 2, 2, 2], [7, 7, 7, 7, 7], [3, 3, 3, 3, 3], [8, 8, 8, 8, 8] ... [0, 0, 0, 0, 0]]

如果可能我不想使用任何循环。

Answer 1

您可以像这样在 numpy 中执行此操作：

import numpy as np

A = np.array([[1, 1, 1, 1, 1], [2, 2, 2, 2, 2], [3, 3, 3, 3, 3], [4, 4, 4, 4, 4], [5, 5 ,5 ,5, 5]])

B = np.array([[6, 6, 6, 6, 6], [7, 7, 7 ,7 ,7], [8, 8, 8, 8, 8], [9, 9, 9, 9, 9], [0, 0, 0, 0, 0]])

print np.hstack((A, B)).reshape((-1, A.shape[1]))

请注意，我已使语法对创建数组有效。

这会产生：

 [[1 1 1 1 1]
 [6 6 6 6 6]
 [2 2 2 2 2]
 [7 7 7 7 7]
 [3 3 3 3 3]
 [8 8 8 8 8]
 [4 4 4 4 4]
 [9 9 9 9 9]
 [5 5 5 5 5]
 [0 0 0 0 0]]

这里发生的事情是np.hstack 创建了一个中间数组，如下所示：

print np.hstack((A, B))

[[1 1 1 1 1 6 6 6 6 6]
 [2 2 2 2 2 7 7 7 7 7]
 [3 3 3 3 3 8 8 8 8 8]
 [4 4 4 4 4 9 9 9 9 9]
 [5 5 5 5 5 0 0 0 0 0]]

这是A 和B 的水平串联。

现在，我将 reshape 使用 .reshape((-1, A.shape[1])) 放入最终矩阵。因为 numpy 数组默认在row-major order 中，所以最终数组中的每一对相邻行都是由该中间数组的一行形成的。

（另见解决方案here）