1. 程式人生 > >python基礎--列表,字典,集合生成式,生成器

python基礎--列表,字典,集合生成式,生成器

列表生成式

需求1:
1.接收變數k,a,b

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

需求2:生成一個列表,列表元素分別為[11,22,33,44…n*n]

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

[experssion for item in 序列 if 判斷語句]

找出1~10之間的所有的偶數

print([i for i in range(1, 11) if i % 2 == 0])

找出1~1000之間的所有質數

 def isPrime(num):
     pass
 print([i for i in range(1, 1000) if isPrime(i)])
 s1='ABC' s2='123'(for巢狀迴圈)
 'A1', 'A2', 'A3', 'B1', 'B2', 'B3', 'C1', 'C2', 'C3'
print([i + j for i in 'ABC' for j in '123'])

列表生成式之for迴圈巢狀

需求:講3x3的矩陣轉換成一堆陣列
[
[1,2,3],
[4,5,6],
[7,8,9]
]
[1,2,3,4,5,6,7,8,9]
“”"

li = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
resLi = []
for item1 in li:  # [1,2,3] [4,5,6] [7,8,9]
    for item2 in item1:
        if item2 % 2 == 0:
            resLi.append(item2)
print(resLi)

列表生成式

   print([item2 for item1 in li for item2 in item1])

匯入模組

 from  itertools import chain
 print(list(chain(*li)))

列表生成式案例

將列表中所有內容都變成小寫

 li = ['dasdaFSDFSFDSF','dadDSADSAsdaDASsadas']
 print([i.lower() for i in li])

找出/var/log目錄中,所有以.log結尾的檔名或者目錄名

import os
#print(os.listdir('/var/log'))
print([filename for filename in os.listdir('/var/log') if filename.endswith('.log')])

列表生成式練習

import math

1.找出1~10之間所有偶數, 並且返回一個列表,(包含以這個偶數為半徑的圓的面積)
方法一:

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)])

方法三:

def square(r):
    """求以r為半徑的圓"""
    res = math.pi *r*r
    return res

print([square(r) for r in range(2,11,2)])

2.找出1~100之間所有的質數.
“”"
判斷num是否為質數,如果是質數,返回True,否則返回False
什麼是質數?
只能被1和本身整除的數,就是質數
方法:
依次判斷num能被(2,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)])

在這裡插入圖片描述

列表生成式練習

給定一個正整數,編寫程式計算有多少對質數的和等於輸入的這個正整數,並輸出結果。輸
入值小於1000。
如,輸入為10, 程式應該輸出結果為2。(共有兩對質數的和為10,分別為(5,5),(3,7))
[2,3,5,7]

輸入描述:
輸入包括一個整數n,(3 ≤ n < 1000)

輸出描述:
輸出對數
示例1 :
輸入:
10
輸出:
2

num=int(input('請輸入一個3~1000的正整數:'))
def isprimer(n):
    for i in range(3, n):
        if n % i == 0:
            return False
    else:
        return True


li = [i for i in range(3, num) if isprimer(i)]
#print(li)
resultli = [(i, j) for i in li for j in li if i + j == num and i<=j]
print(resultli)
print(len(resultli))

字典生成式

需求1:假設有20個學生,學生分數在60~100之間,篩選出成績在90分以上的學生

import random

 stuInfo={}
 for i in range(20):
     name = 'westos' + str(i)
     score = random.randint(60,100)
     stuInfo[name] = score
 print(stuInfo)

字典生成式

 stuInfo = {'westos'+ str(i):random.randint(60,100) for i in range(20)}

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

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

需求2:將所有的key值變為大寫

 d = dict(a=1,b=2)
 new_d = {}
 for i in d:
     new_d[i.upper()] = d[i]
 print('key轉化為大寫的字典:',new_d)

#字典生成式

 print({k.upper():v for k,v in d.items()})

需求3:大小寫key值合併,統一以小寫輸出

d = dict(a=2, b=1, c=2, B=9, A=10)
print({k.lower(): d.get(k.lower(), 0) + d.get(k.upper(), 0) for k in d})
 for k, v in d.items():
     low_k = k.lower()
     if low_k not in new_d:
         new_d[low_k] = v
     else:
         new_d[low_k] += v
 print(new_d)

集合生成式

print({i ** 2 for i in {1, 2, 3}})
print({i ** 2 for i in {1, 2, 3, 9, 12} if i % 3 == 0})

生成器

1.建立生成器的第一重方式

In [7]: nums = (x*2 for x in range(10))

In [8]: nums
Out[8]: <generator object <genexpr> at 0x7fb2a6137fc0>

In [9]: for num in nums:
   ...:     print(num)
   ...:
0
2
4
6
8
10
12
14
16
18

2.建立生成器的方法2(定義一個函式,讓這個函式變成生成器)

 1 1 2 3 5 8 13...
 def fib(num):
     a,b,count = 0,1,1
     while count <= num:
         print(b)
         a,b = b,a+b
         count +=1

 fib(1000)

“”"
如果在呼叫函式的時候,發現這個函式中有yeild
那麼此時,也就不是呼叫函數了,而是建立了一個生成器物件
“”"

def creat_num(all_num):
    print('~~~~~~~~~~~1~~~~~~~~~~~~~~~~~')
    a,b=0,1
    current_num  = 0
    while current_num < all_num:
        print('~~~~~~~~~2~~~~~~~~~~~~~~~~~')
        yield a # 相當於暫停了程式
        #print(a)
        print('~~~~~~~~~~~~~~3~~~~~~~~~~~~')
        a,b = b,b+a
        current_num += 1
        print('~~~~~~~~~~~4~~~~~~~~~~~~~~~~')

obj = creat_num(5)

此時.我們發現程式會報錯(告訴我們,生成器裡面沒有東西了)

while True:
    try:
        ret = next(obj)
        print('obj:',ret)
    except Exception as red:
        print(red.value)
        break

 obj = creat_num(5)
 print(obj)
 ret = next(obj)
 print(ret)
 ret1 = next(obj)
 print(ret1)
 ret2 = next(obj)
 print(ret2)
 ret3 = next(obj)
 print(ret3)
 ret4 = next(obj)
 print(ret4)
 ret5 = next(obj)
 print(ret5)

 for num in obj:
     print(num)
 兩個生成器物件之間並沒有任何關係
 obj2 = creat_num(100)
 print(obj2)
 ret4 = next(obj2)
 print(ret4)

生成器_send

使用send喚醒程式
使用send()函式來喚醒程式執行,使用send()函式的好處是
可以在喚醒的同時向斷點中傳入一個附加的資料

“”"

def create_num(all_num):
    a,b = 0,1
    current_num = 0
    while current_num < all_num:
        ret = yield a
        print('>>>>>>>>ret>>>>>>>',ret)
        a,b = b,a+b
        current_num += 1


obj = create_num(100)
 red = next(obj)
 print(red)
red = obj.send(None)
print(red)

“”"
next和send得到的都是yield後面的值
不同的是send傳遞值而next不傳遞值

注意:
不能把send放在第一個,因為第一次執行程式是從開始執行
並沒有值來接收send
如果你非要把send放在第一個,那麼傳遞的值應該是None
“”"