小编典典

从ND到一维阵列

python

说我有一个数组a

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

array([[1, 2, 3],
       [4, 5, 6]])

我想将其转换为一维数组(即列向量):

b = np.reshape(a, (1,np.product(a.shape)))

但这又回来了

array([[1, 2, 3, 4, 5, 6]])

这与以下内容不同:

array([1, 2, 3, 4, 5, 6])

我可以使用此数组的第一个元素将其手动转换为一维数组:

b = np.reshape(a, (1,np.product(a.shape)))[0]

但这需要我知道原始数组有多少个维数(并在使用较大维数时将[0]连接起来)

有没有一种与尺寸无关的方式来从任意ndarray获取列/行向量?


阅读 54

收藏
2020-12-20

共1个答案: 请登录查看