列表及字典生成式
列表生成式:
列表生成式就是一個用來生成列表的特定語法形式的表達式。
語法格式:
[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()}
總結