决策树划分数据集

这段代码主要是划分数据,比如说看这个矩阵的第一列是否满足需要,如果满足需要,就把后面的添加进来,然后追加到新的矩阵中。。可是这么做有什么用途呢??另外,axis是轴的意思,这段代码给出了三个参数,第一个是要被划分的数据集,第二个是轴线,比如说是第一列,还是第二列,第三个是value,看这一列的数值是否==value
def splitDataSet(dataSet, axis, value):
    retDataSet = []
    for featVec in dataSet:
        print(featVec)
        if featVec[axis] == value:
            reducedFeatVec = featVec[:axis]
            reducedFeatVec.extend(featVec[axis+1:])
            retDataSet.append(reducedFeatVec)
    return retDataSet

featVec[:axis] 是从列表index0到axis,如果axis=0,那么返回的是一个空的列表

featVec[axis+1:]是从列表index axis+1到最后惊恐惊恐

extend 是指将包含的内容加进去

append是将所有的内容加进去,比如一整个列表[1,2,3,5 ]



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值