python - 将 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))