(10)Python——列表、字典、集合生成式

1. 集合生成式

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

    

2.列表生成式

格式：[expression for i in 序列 if ...]

实例：生成一个列表，列表元素分别为[1**2,2**2,3**2,...n**2]

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

#列表生成式
print([i ** 2 for i in range(1,8)])

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

import math #导入数学模块
print([math.pi * i ** 2 for i in range(1,11) if i%2==0])

 结果：

实例3：找出1～100之间所有的质数

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)

实例4：找出.log结尾的文件

import os

print(os.listdir('/var/log'))

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

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

num = int(input('输入一个正整数：'))
#1.先判断2～num之间有多少质数
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
primecount = 0

for item in primeli:
    if (num - item) in primeli and item <= num - item:
        primecount += 1

print('有%s组' %primecount)

 

3.字典生成式

实例1：假设有20个学生，学生名为westosX，学生成绩在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})

实例2：将所有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()})

实例3：大小写、value值和并，统一以小写输出

d = dict(a=2,b=1,c=2,B=9,A=10)
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})