如何优雅生成2的幂次序列及NumPy数组？Python语法糖应用

刚好之前也碰到过类似的需求，这里分享几个Python和NumPy环境下比较简洁优雅的实现方式，供你参考：

Python原生实现 #

固定长度序列 #

如果知道需要生成的序列长度，用列表推导式是最直观简洁的：

# 生成前5个2的幂次：[1, 2, 4, 8, 16]
n = 5
powers = [2**i for i in range(n)]

这里range(n)生成0到n-1的整数，2**i刚好对应从2^0（也就是1）开始的幂次序列，完美匹配你的示例需求。

无限序列（按需生成） #

如果需要一个可以持续生成的无限序列，用itertools.count配合生成器表达式会很方便：

import itertools

# 创建一个无限的2的幂次生成器
power_generator = (2**i for i in itertools.count())

# 按需获取元素，比如取前3个
next(power_generator)  # 1
next(power_generator)  # 2
next(power_generator)  # 4

# 或者一次性取前n个
from itertools import islice
list(islice(power_generator, 5))  # 会得到接下来的5个元素（取决于之前取了多少）

这种方式不需要预先指定长度，适合不确定需要多少元素的场景，而且内存占用极低。

NumPy实现 #

如果是在数值计算场景下，用NumPy的向量操作会更高效，写法也很优雅：

固定长度数组 #

import numpy as np

# 生成前5个2的幂次数组
n = 5
power_array = 2 ** np.arange(n)
# 输出：array([ 1,  2,  4,  8, 16])

这里np.arange(n)生成0到n-1的数组，NumPy支持逐元素的幂运算，一行代码就能搞定，而且处理大规模数据时比原生列表推导式效率高很多。

扩展：生成从某个起始幂次开始的序列 #

如果你的需求不是从1（2^{0）开始，而是比如从2}3=8开始，只需要调整np.arange的起始值：

# 生成从8开始的前4个幂次：[8, 16, 32, 64]
power_array = 2 ** np.arange(3, 7)

内容的提问来源于stack exchange，提问作者Honeybear