Python入门之列表生成式、字典生成式

一、列表生成式

1. 非列表生成式

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

2.列表生成式

s = '51 5000 10000'
li = [int(item) for item in s.split()]
print(li)

练习1：输出[1**1 2**2 .. 9**9]

方法一：

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

方法二：列表生成器

print([i ** i for i in range(1,10)])

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

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

练习3：找出1～n之间所有的质数

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

练习4：给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整数，并输出结果。输入值小于1000。如，输入为10, 程序应该输出结果为2。（共有两对质数的和为10,分别为(5,5),(3,7) [2,3,5,7]

输入描述：输入包括一个整数n,(3 ≤ n < 1000)

输出描述：输出对数

示例1 ： 输入：25   输出：1（2+23）

num = int(input())
def isPrime(num):
    for i in range(2,num):
        if num % i == 0:
            return False
    else:
        return True
primeli = [i for i in range(2,num) if isPrime(i)]
print(primeli)
primecount = 0
for item in primeli:
    if (num - item) in primeli and item <= num - item:
        primecount += 1
print(primecount)

二、列表生成式变形

1. s1 = 'ABC' s2='123'，输出结果为A1 A2 A3 ... C1 C2 C3

print([i + j for i in 'ABC' for j in '123'])

 

2. li  = [ [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]
]
#方法一：
resultli = []
for item1 in li:
    for item2 in item1:
        resultli.append(item2)
print(resultli)
 
#方法二：
print([item2 for item1 in li for item2 in item1])

二 字典生成式

1.假设有20个学生，名字为westosx，学生分数在60-100之间，筛选出成绩在90分以上的学生

方法1：
import random
stuInfo = {}
for i in range(20):
    name = 'westos' + str(i)
    score = random.randint(60,100)
    stuInfo[name] = score
print(stuInfo)
highscore = {}
for name,score in stuInfo.items():
    if score > 90:
        highscore[name] = score
print(highscore)


方法2：使用字典生成器
import random
stuInfo = {}
for i in range(20):
    name = 'westos' + str(i)
    score = random.randint(60,100)
    stuInfo[name] = score
print(stuInfo)
print({name: score for name,score in stuInfo.items() if score > 90})

2. 将所有的key值变为小写

例如：d = dict(a=1,b=2,c=2,B=9,A=10)，将所有key变为小写，并合并相同key的键值（a=11,b=11,c=2)

方法1：
d = dict(a=1,b=2,c=2,B=9,A=10)
print(d)
new_d = {}
for i in d:
    new_d[i.upper()] = d[i]
print(new_d)
new_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)


方法二：字典生成式
d = dict(a=1,b=2,c=2,B=9,A=10)
print({k.lower(): (d.get(k.lower(),0) + d.get(k.upper(),0)) for k in d})