Python-列表推导式

Python 列表推导式

Python中的列表推导式有较好的用途，这里介绍两个可能大家会用到的，具体简单的列表推导式的这里就不说了

testList = [1,2,3,4]
def mul2(x):
    return x*2
print [mul2(i) for i in testList]

还有一张可以通过在内部加上一些逻辑语句从而对数据进行筛选的功能。

def splitDataSet2(dataSet, axis, value):
    temp = [x for x in dataSet if x[axis]==value and x.remove(x[axis])==None]
    return temp

可以看到，我想要去除一个比如[[1,1,0],[1,0,1],[1,0,0],[0,1,1],[0,1,1]]中的每个list的第一个数并返回，这里如果不写一个去除的命令的话式很难做的，这里我把去除的命令放在了判断语句中，当执行判断的时候，进行了操作，这样式非常方便的，不需要另外去写其它函数了，尤其对于这种删除后还不返回列表本身的操作。