如何拒绝项目清单？

Question

我的列表中包含一些值，我想从列表中排除不值并将其替换为0。

那是伪代码：

for i in range(0,img.shape[0]):
for j in range(0,img.shape[1]):
    if img[i][j]!=[some values]:
        img[i][j]=0

Answer 1

这是一种方法：

In [1]: l = [1, 2, 1, 10, 2, 1, 3, 10]

In [2]: exclusions = [1, 2]

In [3]: [el if el not in exclusions else 0 for el in l]
Out[3]: [0, 0, 0, 10, 0, 0, 3, 10]

根据exclusions的大小，如果您将代码转换为set，代码可能会更高效。