python - 将列连接到矩阵(numpy)

Question

我有一个矩阵 b，我需要将其一些列(例如第 2,4 列)放入另一个空矩阵 a 中。基于以下代码:

import numpy as np
a = np.empty([2, 1])
b = np.array([[1, 8, 6, 3], [3, 0, 7, 4]])

所以我想要像这样:

[[8,3][0,4]]

我尝试了np.concatenate:

np.concatenate((a,b[:,1]),axis=0)

但出现此错误:

ValueError: all the input arrays must have same number of dimensions

谁能向我解释一下我应该如何解决这个错误？或者除了 np.concatenate 之外，还有其他方法可以使用吗？

谢谢。

Answer 1

我想你想要的只是

cols = [1, 3]
a = b[:, cols]

您不需要为此目的初始化空矩阵a，也不需要连接。

提醒一下:Python 使用从 0 开始的索引。根据您想要的示例解决方案，您可能正在寻找列 1, 3，而不是 2, 4。

如果您确实确实想要连接(例如，如果a不为空，但您想向其附加更多列，您可以这样做您建议，但沿着 axis=1。但要小心，因为此时您正在将列连接到具有未初始化条目的空矩阵。