python集合的基本操作

python集合的一些基本操作

• len() 获取集合元素个数
• add() 添加元素
• remove() 删除元素
• for...in 遍历集合
• in 判断元素是否在集合中

获取集合的长度

>>> n = {1,2,3,4}
>>> len(n)
4

添加元素

>>> n = {1,2,3,4}
>>> n.add(5)
>>> n
{1, 2, 3, 4, 5}

删除元素

>>> n = {1,2,3,4,5}
>>> n.remove(3)
>>> n
{1, 2, 4, 5}

遍历元素

>>> for attr in n:
...     print(attr)
...
1
2
3
4

判断元素是否在集合中

>>> n = {1,2,3,4}
>>> 1 in n
True
>>> 5 in n
False

集合推导式

使用推导式获取集合中的偶数。

>>> n = {1,2,3,4,5}
>>> v = {number for number in n if number%2 == 0}
>>> v
{2, 4}