Python中列表推导式

1、简单定义

理解：旧的列表通过相应的逻辑生成新的列表。
列表推导式：格式：[表达式 for 变量 in 旧列表] 或者[表达式 for 变量 in 旧列表 if 条件]

2、实例

1. 过滤掉长度小于或者等于3的人名。

   names = ['tom','jason','jack','ha','hey','lilala']
   result = [name for name in names if len(name)>3]  #第一个name表示对name if操作！
   print(result)
   

2. 过滤掉长度小于或者等于3的人名，并且人名的第一个首字母大写。

   names = ['tom','jason','jack','ha','hey','lilala']
   result = [name.capitalize() for name in names if len(name)>3] #第一个name表示对name if操作！
   print(result)
   '''
   #等效于
   def func(names):
   newlist = []
   for name in names:
       name = name.title()
       newlist.append(name)
   return newlist
   '''
   

   注： 第一个name表示对name if操作！

3. 将1-100之间能被3整除，组成一个新的列表

   newlist = [i for i in range(1,101) if i%3 == 0]
   print(newlist)
   

4. 0~5 偶数 0~10奇数。输出如：[(偶数，奇数),(),(),(),()] [(0,1),(0,3),(0,5),(0,7),(0,9),(2,1),(2,3),…]

   newlist = [(x,y) for x in range(5) if x %2 ==0 for y in range(10) if y % 2 !=0]
   print(newlist)
   

5. 薪资大于5000加200，低于或等于5000加500

   dict1 = {'name':'tom','salary':5000}
   dict2 = {'name':'lucy','salary':8000}
   dict3 = {'name':'jack','salary':4500}
   dict4 = {'name':'lily','salary':3000}
   
   list1 = [dict1,dict2,dict3,dict4]
   #if薪资大于5000加200，低于等于5000加500
   newlist = [employee['salary'] + 200 if employee['salary'] > 5000 else employee['salary'] + 500 for employee in list1]
   print(newlist)