arrays - 如何将数组a转换为下面的数组b？

Question

我有以下两个数组 a 和 b。

from numpy import array

a = array([0, 1, 1, 0, 0])

b = array([[0],
           [1],
           [1],
           [0],
           [0]])

如何将数组 a 转换为数组 b？

Answer 1

reshape 方法的替代方法是:

a = np.array([0, 1, 1, 0, 0])
b = a[:,None]
b 
array([[0],
       [1],
       [1],
       [0],
       [0]])

None 将为数组添加一个额外的维度