python - 将 NumPy 向量转换为二维数组/矩阵

Question

将向量转换为二维数组的最佳方法是什么？

例如，大小为 (10, ) 的向量 b

a = rand(10,10)
b = a[1, :]
b.shape

Out: (10L,)

可以转换为大小为(10,1)的数组

b = b.reshape(len(b), 1)

有没有更简洁的方法呢？

Answer 1

由于在使用 a[1, :] 进行索引时丢失了一个维度，因此需要替换丢失的维度以保持 2D 形状。考虑到这一点，您可以使用以下语法进行选择:

b = a[1, :, None]

那么 b 的形状为 (10, 1)。请注意，None 与 np.newaxis 相同，并插入一个长度为 1 的新轴。

(这与编写 b = a[1, :][:, None] 相同，但仅使用一次索引操作，因此节省了几微秒。)

如果你想继续使用 reshape(这也适用于此目的)，值得记住的是你可以对(最多)一个轴使用 -1 让 NumPy 找出正确的长度应该是:

b.reshape(-1, 1)