Python列表推导:将数组中的所有元素设置为0或1

Question

我一直试图想出一个单行列表理解来执行以下操作:给定一个整数数组和一个整数,调用它int1,我想创建一个只有0和1的新数组,这样新的数组如果在原始数组中该位置有一个int1,则为1,否则为0.

有没有办法让真/假为1/0,如同在C？

array1 = [1,4,2,4,5,6,4,3]
array2 = [x == 4 for x in array1 ]
=> [False, True, False, True, False, False, True, False]

Answer 1

简单地将布尔值转换为int,使用int函数,就像这样

array2 = [int(x == 4) for x in array1]

产量

[0, 1, 0, 1, 0, 0, 1, 0]

这是有效的,因为在Python中,Boolean是其子类int.