python中的列表生成式

1.列表生成式练习

形式一：[expression for i in 序列]

1.将字符串里的数字转换为列表输出。

s = '51 5000 10000'

#第一种方法：使用for循环
li = []
for item in s.split():
    li.append(int(item))
print(li)

k,a,b = li
print(k,a,b)

#第二种方法：使用列表生成式
li =[int(item) for item in s.split()]
print(li)

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

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

形式二：[expression for i in 序列 if …]

1.将1到10之间的偶数i的i次方打印出来

# #第一种方法：使用for循环
li = []
for i in range(1,10):
    if i % 2 ==0:
        li.append(i ** i)
print(li)

# #第二种方法：使用列表生成式
li = [i**i for i in range(1,10) if i % 2 == 0]
print(li)

2.找出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)])

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

• 题目描述:
  给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整
  数，并输出结果。输入值小于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)

primecount = 0

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

print(primecount)

形式三：s1 = ‘ABC’ s2=‘123’ A1 A2 A3 … C1 C2 C3
print([i + j for i in ‘ABC’ for j in ‘123’])

1.合并列表

li = [
    [1,2,3],
    [4,5,6],
    [7,8,9]
]

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