列表及字典生成式

列表生成式：

列表生成式就是一个用来生成列表的特定语法形式的表达式。

语法格式：

[exp for iter_var in iter]

迭代iter中的每个元素；

每次迭代都先把结果赋值给iter_var，然后通过exp得到一个新的计算值；

最后把所有通过exp得到的计算值以一个新列表的形式返回。

相当于这样的过程：

L=[ ]

for iter_var in iter:

    a.appond(exp)

 

for语句的列表生成式

例1：

源代码：
 s = '51 5000 10000'
 li = []
 for item in s.split():
     li.append(int(item))
 k,a,b = li

print k,a,b

可用如下代码代替：
 k, a, b = [int(item) for item in s.split()]
 print k,a,b

例2：

要求：生成一个列表，列表元素分别为[1**2,2**2,3**2,4**2...n**2]
源代码：

li = []
 for i in range(1, 8):
      li.append(i ** 2)
 print li

可用如下代码代替：
 print [i ** 2 for i in range(1, 8)]

例3：

要求：找出1~10之间的所有偶数，并且返回一个列表(包含以这个偶数为半径的圆的面积)

源代码：

import math
li = []
for r in range(2, 11, 2):
    square = math.pi * r * r
    li.append(square)
print li

可代替为：

print [math.pi * r * r for r in range(2, 11, 2)]

例4：

要求：找出1～10之间的所有奇数，并且返回一个列表(所有的奇数转换为字符串)
print [str(r) for r in range(1,11,2)]

if语句的列表生成式

例5：
 [experssion for i in 序列 if...]

例:
 print [i ** 2 for i in range(1, 8) if i % 2 == 0]

if语句及for语句的综合

例6：

要求：找出1～num之间的所有质数

源代码：
def isPrime(num):
    for i in range(2,num):
        if num % i == 0:
            return False
    else:
        return True

可代替为：

print [i for i in range(2,101) if isPrime(i)]

字典生成式

例1：

要求：假设有20个学生，成绩在60～100之间，筛选出成绩在90分以上的学生

源代码：
import random

 stuInfo = {}
 for i in range(20):
     name = 'westos' + str(i)
     score = random.randint(60,100)    #随机生成20个60～100之间的成绩
     stuInfo[name] = score
     #以上三条代码可以用下面一条代替

     #stuInfo['westos'+str(i)] = random.randint(60,100)

 hightscore = {}
 for name,score in stuInfo.items():
     if score > 90:
         hightscore[name] = score
 print hightscore

可代替为：

import random

stuInfo1 = {'westos'+ str(i):random.randint(60,100) for i in range(20)}
print {name:score for name,score in stuInfo1.items() if score >90}

例2：

要求：将字典中的key值和value值调换
d = {'a':'apple','b':'bob','c':'cat'}
print {v:k for k,v in d.items()}

d = {'a':'apple','b':'bob','c':'cat'}
print {k:k.upper()for k,v in d.items()}

例3：

要求：大小写合并，key值最终全部为小写
d1 = {'A':10,'b':3,'a':5,'B':8,'d':1}
print {k.lower():d1.get(k.upper(),0)+d1.get(k.lower(),0)for k,v in d1.items()}