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

Question

我有一个 numpy ndarray，在 1 个暗淡的范围内有 6 个值。 这 6 个值代表 6 个维度中的一个点。 如何在 numpy 中将其转换为 6d 数组？

背景故事： 我的神经网络需要这个 6d 布局。训练发生在 6d 数据上。为了做出预测，我未能重塑数据。

Answer 1

您可以使用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))

Answer 2

这里是您在上面的评论中提到的要求的解决方案，但重要的是要了解 (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]