如何根据另一个列表中的值从列表中删除元素？

Question

例如，我有两个列表： nums = [1,1,2] not_delete=[1,0,1] 我想从 nums 中删除中间的 1 并返回 [1, 2]，因为 not_delete 数组中的相应值为 0。 在python中有没有一种直接的方法可以做到这一点？

Answer 1

建立一个新列表，只保留您想要的列表。

nums = [n for i,n in enumerate(nums) if not_delete[i]]

Numpy 可以在一个语句中完成此操作，如果您使用的是 numpy。