featList = [example[i] for example in dataSet]

该语句是将列表list，取一列元素数据的方法。
运营原理为：先提取一行dataSet的数据，然后取该行数据的第“ i ”位元素。然后遍历每一行，最后获得一整列数据，变为一个列表list。

dataSet=[[1,1,'yes'],
         [1,1,'yes'],
         [1,0,'no'],
         [0,1,'no'],
         [0,1,'no']]
featList1=[example[0] for example in dataSet]
print('featList1 = \n',featList1)
print(type(featList1))
>>> featList1 = 
>>> [1, 1, 1, 0, 0]
>>> <class 'list'>

featList2=[example[-1] for example in dataSet]
print('featList2 = \n',featList2)
>>> featList2 = 
>>> ['yes', 'yes', 'no', 'no', 'no']
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容