Python之列表推导式

2020-12-20

列表推导式是Python语言的特性之一。列表推导式的基本形式如下：

[expr for val in collection if condition]

如果过滤条件省略,只保留表达式的形式如下：

[expr for val in collection]：

1
2
3

strs = ['a','as','apple','bat','car','dove','cplusplus']
up_2 = [x.upper() for x in strs if len(x) > 2]
print(up_2)

集合与字典的推导式是列表对导师的自然拓展，其形式如下：

dict_comp={key-expr: value-expr for value in collection if condition}

set_comp = {expr for value in collection if condition}

和列表推导式类似，集合，字典的推导式非常方便。

unique_len = {len(s) for s in strs}
loc_mapping = {key : index for index,key in enumerate(strs)}
print(loc_mapping)
print(set(map(len,strs)))
print(unique_len)

列表推导式也是可以嵌套使用的，要牢记for表达式的顺序应当和你写嵌套for循环来代替列表推导式的顺序一致。嵌套列表推导式和列表推导式中的列表推导式不一样。

嵌套列表推导式：

1
2
3

tuples = [(1,2,3),(4,5,6),(7,8,9)]
flattend = [x for tup in tuples for x in tup]
print(flattend)

列表推导式中的列表推导式：

1 2	dim2 = [[x for x in tup ] for tup in tuples] print(dim2)