python-生成式的基本使用

Posted 2020-11-24 liuxu2019

生成式是python中的一种高级玩法，起码看起来显得要高级一点。它可以使用简单的一行代码实现列表、字典等数据类型的创建或数据类型的转换等任务。另外，它和生成器还有些许关联。

列表生成式

列表生成式即生成列表的生成式，写法简单而优雅，可以将多行代码用一行代码实现，且效率较高。一般用于创建简单的新列表或对过滤出新列表。另外，还可以做数据类型间的转换。

# 方式1
nums = [i for i in range(10) if i%2 == 0]       # nums= [0, 2, 4, 6, 8]
nums2 = [i for i in nums if i >5]               # num2= [6, 8]

# 方式2
nums = list(i for i in range(1, 5))             # nums = [1, 2, 3, 4]

# 数据类型转换
a = {"name":"jack", "age": 18}
a2 = [key for key in a]                         # a2 = ['name', 'age']

字典生成式

字典生成式即生成字典的生成式。特殊场合下，可以写出很优美的代码。也分为两种生成方式，一种默认的使用{},括号体内使用循环生成。另外配合内置方法的使用，效果很优美。

# 方式1
{k:v for k,v in enumerate(range(1,5))}  # {0: 1, 1: 2, 2: 3, 3: 4}

# 方式2
li = [[1,2], (2,3)]
dict(x for x in li)         # {1: 2, 2: 3}

dict(zip([1,2], [3,4]))     # {1: 3, 2: 4}

集合生成式

# 方式1
a = set(i for i in range(1,5))      # {1,2,3,4}
# 方式2
b = {i for i in range(1,5)}         # {1,2,3,4}

元组生成式

元组生成式即生成元组的生成式，按理说可以使用()，括号内循环的方式生成元组生成式，但是()被python中的生成器占用，就只剩下一种方式生成元组生成式。

# 方式1
tuple(x for x in range(1,5))        # (1, 2, 3, 4)

# 数据类型转换
nums = [3,4,5,6]
tuple(x for x in nums)              # (3, 4, 5, 6)

生成器生成式

使用()内循环的方式可以生成最简单的生成器，生成器替代列表的使用可以减小内存的消耗。生成器可以通过 next输出数据，但最后一个后会报错；一般使用for循环的方式使用生成器。其实，range(10)本质上就是一个最简单的生成器。

a = (i for i in range(1,5))     # <generator object <genexpr> at 0x0000027B1B5994C8>

print(next(a))      # 1
print(next(a))      # 2
print(next(a))      # 3
print(next(a))      # 4
print(next(a))      # 抛出异常StopIteration

# 可以使for循环的方式避免异常
print([i for i in a])   # [1,2,3,4]