numpy快速入门

Posted 2021-04-04 chenxi188

列表推导式（非numpy部分）

1.首先看看常规写法，这非常麻烦

#常规写法
>>> myList=[]
>>> for item in a:
... myList.append(item*4)
...
>>> myList
[4, 8, 8, 8, 16, 20, 20]

2.列表推导式写法，上式等于

#列表推导式是这样子滴：
>>> a=[1, 2, 2, 2, 4, 5, 5]
>>> myList = [item*4 for item in a]
>>> myList
[4, 8, 8, 8, 16, 20, 20]

3.列表推导更高级写法，只保留大于2的写法

>>> a=[1, 2, 2, 2, 4, 5, 5]
#只取原列表大于2的，来计算，其它不要
>>> [item*4 for item in a if item>2] 
[16, 20, 20]