python - 在python中的集合函数中消除0

Question

如何从集合函数中消除 0

uniq_events = set((row[self.param]) for row in self.fpt)
o/p
set(['1','2','0','3')

现在我想要一个像这样的o/p

set(['1','2','3')

Answer 1

我只想在 uniq_events 包含None或0在集合中时消除。

你可以使用evt -= {'0', None}：

In [10]: evt = set(['1','2','0','3',None])

In [11]: evt
Out[11]: set(['1', '0', '3', '2', None])

In [12]: evt -= {'0', None}

In [13]: evt
Out[13]: set(['1', '3', '2'])

或者，您可以在填充集合时删除不需要的元素：

uniq_events = set(row[self.param] for row in self.fpt if row not in [None, '0'])

Answer 2

您可以就地完成：

uniq_events.discard("0")

或创建一个新集：

new_events = uniq_events - {"0"}