将6值的1维numpy.ndarray转换为6维

Question

我有一个numpy ndarray，其中1个昏暗处有6个值。 这6个值代表6个维度中的一个点。 如何将其转换为numpy以获得6d数组？

背景故事： 我的神经网络需要这个6d布局。训练使用6d数据进行。为了制作谓词，我无法重塑数据。

Answer 1

这里是您在注释中提到的您的需求的解决方案，但是要理解的重要一点是(6,)不是6-dimensional数组。它是形状为(6, )且为1-dimensional且由6元素组成的数组。

import numpy as np 
x = np.array([1,2,3,4,5,6]) 
x = np.reshape(a, (6,))
print(x.shape)

输出：

(6,)

print(x)

输出：

[1 2 3 4 5 6]

Answer 2

您可以使用numpy.reshape

import numpy as np
# for example my numpy array having 6 elements is [1,2,3,4,5,6]
a = np.array([1,2,3,4,5,6])
# array([1, 2, 3, 4,5,6])
b = np.reshape(a, (-1,1))