假设我有一个数组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 获取列/行向量的与维度无关的方法?