Python中处理空列表问题：map映射空列表时的报错解决

问题复现 #

当列表A、B均为空时，执行以下代码会抛出ValueError：

A=[]
B=[]
tol=1e-12

CA, CB = map(list, zip(*((a, b) for a, b in zip(B, A) if a[0]>tol)))

print(CA)
print(CB)

报错信息：

ValueError: not enough values to unpack (expected 2, got 0)

期望输出：

CA=[]
CB=[]

报错原因 #

当A、B为空时，zip(B, A)生成空迭代器，过滤条件a[0]>tol自然不会匹配任何元素，最终传给zip(*...)的是一个空生成器。zip(*空生成器)返回空迭代器，map(list, 空迭代器)同样没有输出，此时尝试将空迭代器解包为两个变量（CA、CB），就会因“没有足够的值”抛出错误。

解决方案 #

通过条件判断处理空结果的情况，先捕获过滤后的结果，再决定后续操作：

A=[]
B=[]
tol=1e-12

# 先将生成器转为列表，便于判断是否为空
filtered = list((a, b) for a, b in zip(B, A) if a[0]>tol)
# 若过滤结果为空，直接赋值空列表；否则执行原逻辑
CA, CB = map(list, zip(*filtered)) if filtered else ([], [])

print(f"CA={CA}")
print(f"CB={CB}")

或者更简洁的写法（利用空迭代器在布尔判断中为False的特性）：

A=[]
B=[]
tol=1e-12

CA, CB = map(list, zip(*((a, b) for a, b in zip(B, A) if a[0]>tol))) or ([], [])

print(f"CA={CA}")
print(f"CB={CB}")

验证结果 #

运行修正后的代码，会输出期望的：

CA=[]
CB=[]

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