将数组转换为0和1

Question

假设我有一个数组A = [13, 15, 17]。我想创建一个新的数组B，这样除了它的第13、15和17个条目以外的所有条目都是0，而这三个条目中的每一个都是1。我该怎么做？你知道吗

Answer 1

使用列表理解：

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

对于从0到A的最大值的每个数字，我们取int(i+1 in A)。i+1 in A将是一个布尔值。如果该数字在A中，则结果将是True。否则就是False。由于bool继承自int，我们可以很容易地用int()将其转换为一个普通整数。你知道吗