python 生成式

以下所有的实操全在redhat7.3真机上pycharm

---

列表生成式

描述:

列表生成式即List Comprehensions，是Python内置的非常简单却强大的可以用来创建list的生成式

语法：

[expression for i in 序列 if...]

s = '51 5000 10000'
#print(s.split(),type(s.split()))   #s.split()分离后是一个列表

#普通表达式
li = []
for item in s.split():
    li.append(int(item))
k,a,b = li
#列表生成式表达
k,a,b = [int(item) for item in s.split()]   #列表生成式 ，遍历之后直接生成在列表当中；
#           ^返回值       ...循环语句
print(k,a,b)

练习：生成一个列表，列表元素分别为[12,22,32…n2]

#普通表达式
li = []
for i in range(1,10):
    li.append(i**2)
print(li)
#列表生成式表达
print([i**2 for i in range(1,10)])

#偶数平方列表

print([i**2 for i in range(1,10) if i%2 == 0])
#       ^满足         (1)                    (2)
#      (1)(2)条件则返回值i	

练习：1~num之间所有的质数：除了1和自身被整除以外，无其他数可整除，该数为质数

def isPrime(num):
     for i in range(2,num):
        if num % i == 0:
            return False  #当检测到有一个（除了1和自身的数）可以整除，该数就不为质数，结束循环，其实也可以使用break
     else:
            return True
print([i for i in range(2,101) if isPrime(i)])
                                #^该判断句成立时返回该数i

练习题：

(2018-腾讯-在线编程题)

• 题目描述:
  给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整
  数，并输出结果。输入值小于1000。
  如，输入为10, 程序应该输出结果为2。（共有两对质数的和为10,分别为(5,5),(3,7)）
  #[2,3,5,7]
• 输入描述:
  输入包括一个整数n,(3 ≤ n < 1000)
• 输出描述:
  输出对数
  -> 示例1 :
  输入:
  10
  输出:
  2

#计算出num以内的所有的质数：
num = int(input())
if num > 1000:
    exit()
else:
    def isPrime(num):
        for i in range(2, num):
            if num % i == 0:
                return False
        else:
            return True
    li = [i for i in range(2,num) if isPrime(i)]   #将所有的质数放入其列表中
    # print(li)
#1.先拿出两个数
#2.判断这两个数相加是否等于num
count = 0
for item in li:    #item是将li列表中的所有元素遍历一遍
    if (num - item) in li and item <= num-item:
        count += 1
print(count)

---

字典生成式

需求：假设有20个学生，学生名为westisX，学生分数在60-100之间
#选出成绩在90分以上的学生

代码：

import random
studentInfo = {}
for i in range(20):
    name = 'westos' + str(i)
    score = random.randint(60, 100)
    studentInfo[name] = score
 #print(studentInfo)
#普通表达式
highscore = {}
for name,score in studentInfo.items():
    if score > 90:
        highscore[name] = score
print(highscore)
#列表表达式
print({name:score for name,score in studentInfo.items() if score > 90})

运行结果：

需求：将所有的key值变为大写

代码：

d = dict(a=1,b=2)
#普通表达式
new_d = {}
for i in d:
    new_d[i.upper()] = d[i]   #d[i]：根据键找值；
print(new_d)
#列表表达式
print({k.upper():v for k,v in d.items()})

运行结果：
#需求：大小写key的值和并，统一以小写输出
代码：

d = dict(a=2,b=1,c=7,C=10,A=17)
#{'a':2,'b':1,'c':7,'C':10,'A':17}

#普通生成式
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)
get()函数
#字典(Dictionary) get() 函数返回指定键的值，如果值不在字典中返回默认值None。
#语法：dict.get(key, default=None)
#参数：key -- 字典中要查找的键。；default -- 如果指定键的值不存在时，返回该默认值值。

#列表生成式
print({k.lower():d.get(k.lower(),0) + d.get(k.upper(),0) for k in d})
#这条语句中get(key,0)是当键不存在的话，返回值0；

---

生成式变式

li = [
    [1,2,3],
    [4,5,6],
    [7,8,9]
]
#
##将列表中的列表变成一个列表[1,2,3,4,5,6,7,8,9]
#普通生成式
result = []
for i in li: #[1,2,3],[4,5,6],[7,8,9]
    for j in i:
        result.append(j)
print(result)
#列表生成式
print([j for i in li for j in i])


#利用迭代器(itertools)方法
from itertools import chain

#chain函数来自于itertools库，itertools库提供了非常有用的基于迭代对象的函数，
#而chain函数则是可以串联多个迭代对象,来形成一个更大的迭代对象
print(list(chain(*li)))   #*：多个