如何用无NumPy的Python简洁语法按条件生成数组副本

最直接的一行写法（列表推导式） #

如果a是原生Python列表，用列表推导式就能直接生成符合要求的副本，这是最常用的简洁写法：

a = [2, 7, -2, 0, 0, 9]
b = [1 if x != 0 else 0 for x in a]
print(b)  # 输出: [1, 1, 1, 0, 0, 1]

符合“包含两个等号”的写法（Python 3.8+） #

你记忆里带两个等号的写法，可以用Python 3.8引入的海象运算符（:=）实现，形式上和你说的接近：

a = [2, 7, -2, 0, 0, 9]
b = [1 if (val := x) != 0 else 0 for x in a]
print(b)  # 输出: [1, 1, 1, 0, 0, 1]

这里val := x是赋值表达式，加上后面的!= 0，刚好包含两个“等号”，满足你提到的形式要求。

其他简化思路 #

• 利用布尔值转整数：非零值的布尔判断结果是True（转整数为1），零是False（转整数为0），可以简化成：

b = [int(bool(x)) for x in a]

• 用map配合lambda：

b = list(map(lambda x: 1 if x != 0 else 0, a))

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