Numpy Arrays，形状错误

Question

我正在学习入门级 ML 课程，并且正在分析 sklearn 库中的 iris 数据集，我需要做的一件事是打印出与数据列表关联的第三个特征。 我还需要确保形状是 (150,)。

我尝试了什么：

x = iris.data[:, 2:3]这确实给了我第三个特征，但形状是 (150,1) 而不是 (150,) 我不太明白为什么。

将不胜感激一些帮助，在此先感谢。

Answer 1

iris.data[:, 2:3]为您提供功能 2（包括）到 3（不包括）的视图，将其视为 1 个功能的列表，因此是您形状中的最后一个1 。 只需直接使用iris.data[:, 2]使其成为一维数组（其形状的长度为 1）。