Python——列表生成式

概念

• 列表生成式即List Comprehensions，是Python内置的非常简单却强大的可以用来创建list的生成式。
• 格式：[表达式 for i in 序列 if...]

示例

如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做？

• 方法一是循环：

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

• 方法二：使用列表生成式

li1 = [i*i for i in range(1,11)]
print(li1)

如果要找到指定范围内的所有偶数？

• 原方法：

li = []

for i in range(11):
    if i % 2 == 0:
        li.append(i)
print(li)

结果：

• 使用列表生成式：

print([i for i in range(11) if i % 2 == 0])

可以很直观的发现，使用列表生成式非常简洁。

统计10以内的半径为偶数的圆的面积

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

• 方法1：利用条件判断

import math

print([math.pi * i ** 2 for i in range(1,11) if i % 2 == 0])

结果：

• 方法2：利用步长

import math

print([math.pi * i ** 2 for i in range(2, 11, 2)])
print([math.pi * i ** 2 for i in range(1,11) if i % 2 == 0])

结果：

找到n以内的所有素数

def isPrime(num):
    # 0和1 不是质数，特殊处理
    if num == 0 or num == 1:
        return False
    # 除了1和它本身，如果可以整除，就不是素数
    for i in range(2,num):
        if num % i == 0:
            return False
    else:
        return True

print([isPrime(i) for i in range(10)])
print([i for i in range(10) if isPrime(i)])

结果：

腾讯笔试题：找质数对和等于目标数

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

• 思路：题目说目标数字=质数之和，那么
  1.首先得在目标数字范围内找到所有的质数，存到列表里。
  2.其次，用 目标数字 - 质数 是否在质数的列表里来判断，有的话计数器+1
  3.全部遍历质数列表的话，会多存一遍质数对，例如3+7 和 7+3 所以遍历质数列表的一半即可

def isPrime(num):
    if num == 0 or num == 1:
        return False
    for i in range(2, num):
        if num % i == 0:
            return False
    else:
        return True


intNum = int(input())
# 利用列表生成式，存输入数字以内所有的质数
primeli = [i for i in range(intNum) if isPrime(i)]

resCount = 0

for i in primeli:
    # 判断 输入数-指数列表里的数是否是质数 同时 为了避免重复，必须这个数要小于等于另一个质数
    if intNum - i in primeli and i <= intNum - i:
        resCount += 1
print(resCount)

结果：

列表生成式变形

• 列表生成式嵌套for循环：

s1 = 'ABC'
s2 = '123'
#   嵌套循环方法一：
li = []
for i in s1:
    for j in s2:
        li.append(i+j)
print(li)

#   嵌套循环方法二：
print([i+j for i in s1 for j in s2])

结果：

• 将嵌套列表元素整合到一个列表中

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

# 目标是将列表转换成[1,2,3,4,5,6,7,8,9]

# 1.原始方法
resli = []
for i in li:
    for j in i:
        resli.append(j)
print(resli)
# 2.使用列表生成式
print([j for i in li for j in i])

结果：