在numpy中將向量組合為列矩陣

Question

我有以下三個向量：

a = np.ones(20)
b = np.zeros(20)
c = np.ones(20)

我正在嘗試將它們合並為20x3尺寸的矩陣。

目前我正在做：

n1 = np.vstack((a,b))
n2 = np.vstack((n1,c)).T

這可行，但是沒有辦法以列方式用數組填充矩陣嗎？

Answer 1

您可以通過幾種不同的方法來執行此操作。 以下是一些示例：

使用np.c_ ：

np.c_[a, b, c]

使用np.dstack和np.squeeze ：

np.dstack((a, b, c)).squeeze()

使用np.vstack和轉置（類似於您的方法）：

np.vstack((a,b,c)).T

使用np.concatenate和reshape ：

np.concatenate((a, b, c)).reshape((-1, 3), order='F')

如果效率很重要，那么使用np.concatenate的最后一種方法似乎是我計算機上最快的方法：

>>> %timeit np.c_[a, b, c]
10000 loops, best of 3: 46.7 us per loop

>>> %timeit np.dstack((a, b, c)).squeeze()
100000 loops, best of 3: 18.2 us per loop

>>> %timeit np.vstack((a,b,c)).T
100000 loops, best of 3: 17.8 us per loop

>>> %timeit np.concatenate((a, b, c)).reshape((-1, 3), order='F')
100000 loops, best of 3: 3.41 us per loop

Answer 2

是的，使用column_stack ：

np.column_stack((a,b,c))

這適用於堆疊通用的一維數組。 在您想要20x3數組，使得每一行為(1,0,1)的特定情況下，我建議：

np.tile([1.,0.,1.], (20,1))