Python 将数组转换为0和1

假设我有一个数组

A=[13,15,17]

。我想创建一个新数组

B

，这样除了第13、15和17个条目之外，所有条目都是

0

，这三个条目都是

1

。我该怎么做

使用列表理解：

B = [int(i+1 in A) for i in range(max(A))]

---

对于从

0

到

A

中的最高数字范围内的每个数字，我们取

int（A中的i+1）

中的code>i+1将是一个布尔值。如果该数字在

A

中，则结果将为

True

。否则它将为

False

。由于

bool

继承自

int

，因此我们可以使用

int（）

假设a=[5]，轻松地将其转换为正常整数。我希望B是[0,0,0,0,1]