我正在参加一个入门级的机器学习课程,目前在分析来自sklearn库的鸢尾花数据集。我需要做的一件事是打印出数据列表中与第三个特征相关的信息。同时,我还需要确保数据的形状是(150,)。
我尝试过的方法:
x = iris.data[:, 2:3]
确实能给我第三个特征,但形状是(150,1)而不是(150,),我不太明白为什么会这样。
希望能得到一些帮助,提前感谢。
回答:
iris.data[:, 2:3]
给你的是从特征2(包括)到特征3(不包括)的视图,可以认为这是一个包含一个特征的列表,所以形状中的最后一个1
就是由此而来。直接使用iris.data[:, 2]
可以得到一个一维数组(其形状长度为1)。