如何高效且易读地利用布尔列表将列表元素插入指定位置？

嘿，这个需求我太熟悉了！你当前的写法确实能实现功能，但确实可以更简洁高效，同时保持良好的可读性。这里给你推荐两种不错的实现方式：

纯Python简洁实现（推荐） #

利用迭代器的特性，我们可以直接遍历ind列表，遇到True就取出a的下一个元素，否则填充0，用列表推导式一步完成：

a = [10, 200]
ind = [False, True, False, True, False]

# 创建a的迭代器，每次next会按顺序取a的元素
a_iter = iter(a)
b = [next(a_iter) if flag else 0 for flag in ind]

这样写的好处：

• 代码紧凑易读，一眼就能看明白逻辑
• 只需要遍历一次ind，时间复杂度为O(n)（n是ind的长度），效率很高
• 不需要额外生成positions中间列表，节省内存

运行后b的结果就是[0, 10, 0, 200, 0]，完全符合你的需求。

大数据场景：用numpy快速实现 #

如果你的数据量比较大，或者本身就在用numpy做数据分析，那可以用numpy的布尔索引来实现，代码更简洁：

import numpy as np

a = np.array([10, 200])
ind = np.array([False, True, False, True, False])

# 初始化全0数组
b = np.zeros(len(ind), dtype=int)
# 直接通过布尔索引赋值
b[ind] = a

numpy的底层是C实现的，这种赋值方式在处理大规模数据时效率会比纯Python更高，而且代码非常直观。

对比你的原有实现 #

你原来的写法需要先生成positions列表，再循环赋值，多了一次遍历和中间列表的生成。上面两种方法都省去了这一步，代码更简洁，逻辑也更直接。

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