python推导式

127 阅读1分钟

1.列表推导式

for循环的运用

list0 = [i for i in range(5)] 
print(list0)

输出结果为:
[0, 1, 2, 3, 4]

添加if语句

list1 = [i for i in range(5) if i != 3] 
print(list1)

输出结果为:
[0, 1, 2, 4]

2.字典推导式

如下例子(将两个独立的列表组成为一个字典,一个列表里的值全部为键,另一个为值)

list1 = ['a','b','c'] 
list2 = [10,20,30]
dict = {list1[i]: list2[i] for i in range(len(list1))} 
print(dict)

输出结果为:
{‘a’: 10, ‘b’: 20, ‘c’: 30}

3.集合推导式

list = [1,1,1,5]
set1 = {i + 1 for i in list} 
print(set1)

输出为:
{2, 6}(集合自动去重)