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）。