python学习笔记——列表生成式的使用

列表生成式12个案列

前言

Python 里使用 [] 创建一个列表。容器类型的数据进行运算和操作，生成新的列表最高效的办法——列表生成式。

列表生成式优雅、简洁，值得多多使用！

基础部分

1. 数据再运算

实现对每个元素的乘方操作后，利用列表生成式返回一个新的列表。

In [1]: a = range(0,11)

In [2]: b = [x**2 for x in a] # 利用列表生成式创建列表

In [3]: b
Out[3]: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

数值型的元素列表，转换为字符串类型的列表：

In [1]: a = range(0,10)

In [2]: b = [str(i) for i in a]

In [3]: b
Out[3]: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']

2. 一串随机数

生成 10 个 0 到 1 的随机浮点数，保留小数点后两位：

from random import random
a = [round(random(),2) for _ in range(10)]

生成 10 个 0 到 10 的满足均匀分布的浮点数，保留小数点后两位：

from random import uniform
a = [round(uniform(0,10),2) for _ in range(10)]

3. if 和嵌套 for

列表生成式中嵌套if语句：
对一个列表里面的数据筛选，只计算 [0,11) 中偶数的平方：

a = range(11)
c = [x**2 for x in a if x%2==0]
print(c)
>> [0, 4, 16, 36, 64, 100]

列表生成式中嵌套 for语句：
如下使用嵌套的列表，一行代码生成 99 乘法表的所有 45 个元素：

a = [i*j for i in range(10) for j in range(1,i+1)] 
>> 

4. zip 和列表

zip(*iterables)——创建返回一个迭代器，聚合每个可迭代对象的元素。

#zip单独使用
a = [0, 1, 2, 3]
b = ['a', 'b', 'c', 'd']
print(list(zip(a,b)))
>> [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]

#列表生成式结合zip函数使用
a = range(5)
b = ['a','b','c','d','e']
c = [str(y) + str(x) for x, y in zip(a,b)]
print(c)
>> ['a0', 'b1', 'c2', 'd3', 'e4']

5. 打印键值对

a = {'a':1,'b':2,'c':3}
b = [k+ '=' + str(v) for k, v in a.items()]
print(b)
>> ['a=1', 'b=2', 'c=3']

6. 文件列表

import os
a = [d for d in os.listdir('D:/source/test')]

再结合 if，返回文件夹的名字列表：

import os
dirs = [d for d in os.listdir('D:/source/test') if os.path.isdir(d)]
print(dirs)
>> ['.vscode', 'templates', '__pycache__']

只查找出文件。

import os
files = [d for d in os.listdir('D:/source/test') if os.path.isfile(d)]

print(files)
>> ['a.sql','barchart.py','barstack.py','bar_numpy.py']

7. 转为小写

a = ['Hello', 'World', '2019Python']
b = [w.lower() for w in a]
print(b)
>> ['hello', 'world', '2019python']

该写法可能会有问题，因为 Python 的列表类型可以不同，如果列表 a：

a = ['Hello', 'World',2020,'Python']

[w.lower() for w in a]
---------------------------------------------------------------------------
AttributeError: 'int' object has no attribute 'lower'

如上就会出现 int 对象没有方法 lower 的问题，先转化元素为 str 后再操作：

a = [str(w).lower() for w in a]
print(a)
>> ['hello', 'world', '2020', 'python']

更友好的做法，使用 isinstance，判断元素是否为 str 类型，如果是，再调用 lower 做转化：

a = [w.lower() for w in a if isinstance(w,str) ]
print(a)
>> ['hello', 'world', 'python']

8. 保留唯一值

def filter_non_unique(lst):
   return [item for item in lst if lst.count(item) == 1]

filter_non_unique([1, 2, 2, 3, 4, 4, 5])
>> [1, 3, 5]

高级部分

9. 筛选分组

def bifurcate(lst, filter):
    return [
        [x for i,x in enumerate(lst) if filter[i] == True],
        [x for i,x in enumerate(lst) if filter[i] == False]
        ]

b = bifurcate(['beep', 'boop', 'foo', 'bar'], [True, True, False, True])
print(b)
>> [['beep', 'boop', 'bar'], ['foo']]

10. 函数分组

def bifurcate_by(lst, fn):
	return [
		[x for x in lst if fn(x)],
		[x for x in lst if not fn(x)]
		]

b = bifurcate_by(['Python3', 'up', 'users', 'people'], lambda x: x[0] == 'u')
print(b)
>> [['up', 'users'], ['Python3','people']]

11. 差集

def difference(a, b):
	_a, _b =set(a),set(b)
	return [item for item in _a if item not in _b]

b = difference([1,1,2,3,3], [1, 2, 4])
print(b)
>> [3]

12. 函数差集

列表 a、b 中元素经过 fn 映射后，返回在 a 不在 b 中的元素。

from math import floor
def difference_by(a, b, fn):
		_b = set(map(fn, b))
		return [item for item in a if fn(item) not in _b]

#列表元素为单个元素：
b = difference_by([2.1, 1.2], [2.3, 3.4],floor)
print(b)
>> [1.2]

#列表元素为字典：
c = difference_by([{ 'x': 2 }, { 'x': 1 }], [{ 'x': 1 }], lambda v : v['x'])
print(c)
>>  [{'x': 2}]