Python-列表生成式、字典生成式及生成式变式

列表生成式

# s = '51 5000 10000'
# 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)

#生成一个列表，列表元素分别为[1**2,2**2,3**2...n**2]
# li = []
# for i in range(1,10):
#     li.append(i ** 2)
# print(li)

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

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

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

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

列表生成式练习

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

[2,3,5,7]

输入描述:
输入包括一个整数n,(3 ≤ n < 1000)
输出描述:
输出对数
示例1 :
输入:
10
输出:
2

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)

"""
1.先拿出两个数
2.判断两个数之和是否等于num
"""
count = 0

for item1 in primeli:
    if (num - item1) in primeli and item1 <= num - item1:
        count += 1

print(count)

字典生成式

#需求：假设有20个学生，学生名为westisX，学生分数在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)

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

#需求：将所有的key值变为大写
 d = dict(a=1,b=2)
 print(d)

 new_d = {}
 for i in d:
     new_d[i.upper()] = d[i]

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


#需求：大小写key的值和并，统一以小写输出
d = dict(a=2,b=1,c=2,B=9,A=10)
{'a':12,'b':10,'c':2}
 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)

print({k.lower(): d.get(k.lower(),0) + d.get(k.upper(),0)
       for k in d})

生成式变式

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

from itertools import chain

print(list(chain(*li)))