python集合类型

时间:2021-12-26 19:20:49

集合类型简介

集合也是容器,其内元素都是无序、唯一、不可变的。它常用来做成员测试、移除重复数据、数据计算(比如交集、并集、差集)。

集合Set是dict的无value版。集合也使用大括号包围:

>>> s = {'a','b','c'}
>>> type(s)
<class 'set'>

Set中的元素和dict一样也是通过hash值来存储的:将元素hash()得到hash值,存储到Set中。所以,Set中的元素必须是不可变数据(例如列表不能放进集合中)。但集合自身是可变的,可以修改其中的元素。此外,python提供了另一种不可变的集合类型frozenset。

使用大括号或set()构造方法可以构造集合。

s = {'a','b','c'}
s = set("abc")
s = frozenset("abc")

需要注意的是,空的{}表示的字典,而不是集合,如果想要构造空集合,可以使用不带参数的set()来构造。

由于集合是通过hash值来存储的,没有位置索引。所以没法对集合进行单元素的检索,只能对集合进行修改操作,或迭代、遍历。

集合运算操作

>>> x = set("abcde")
>>> y = set("defgh")
>>> z = set("opq")
>>> x
{'b', 'e', 'c', 'a', 'd'}
>>> y
{'e', 'h', 'f', 'g', 'd'}

集合的运算有交集、并集、差集等操作。它们都有两种方式:操作符号版的,方法函数版的。符号版的都只能集合对集合,函数版的可以集合和其它比较,比如列表。

交集&或intersection()方法:

>>> x & y
{'e', 'd'} >>> x.intersection(y)
{'e', 'd'} >>> x & ["a", "c"]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for &: 'set' and 'list' >>> x.intersection(["a","c"])
{'c', 'a'}

并集|或union()方法:

>>> x | y
{'b', 'e', 'h', 'c', 'a', 'f', 'g', 'd'}
>>> x | y | z >>> x.union(y)
>>> x.union(["x", "y"])
>>> x.union(["x", "y"], ["o", "p"])

差集-或difference():

>>> x-y
{'c', 'a', 'b'}
>>> y-x
{'g', 'f', 'h'} >>> x.difference(y)
{'c', 'a', 'b'}
>>> y.difference(x)
{'g', 'f', 'h'}

还有XOR操作,取集合1、集合2中非交集的部分:

>>> x ^ y
{'f', 'a', 'h', 'b', 'c', 'g'} >>> x.symmetric_difference(y)

下面是测试两个集合之间是否是子集、真子集、超集的关系,s1和s2都是集合。同样,使用函数版的可以是其它类型。

# 子集
s1 <= s2
s1.issubset(s2)
s1.issubset(other_type) # 真子集
s1 < s2 # 超集
s1 >= s2
s1.issuperset(s2)
s1.issuperset(other_type) # 真超集
s1 > s2

s1.isdisjoint(other_type)测试集合和另一个数据容器(如集合、列表)是否存在相交数据。即集合中的元素和其它容器是否有共同数据,如果有则返回False,否则返回True。

>>> x.isdisjoint(y)
False >>> x.isdisjoint(z)
True >>> x.isdisjoint(list("ab"))
False
>>> x.isdisjoint(list("opq"))
True

集合基本操作

集合类型(不是frozenset)是可变的容器类型,可以修改它(但没法检索它)、测试、迭代它,但不能检索它(除非迭代、遍历)。

s1.add(elem)添加元素到集合s1中。因为集合中的元素都唯一,所以添加已存在的元素不会有任何效果,但也不会报错。

s1.remove(elem)移除集合s1中的元素。

s1.pop()随机移除一个元素并返回这个元素。

s1.clear()清空集合。

s1.discard(elem)移除已存在的某个元素,如果不存在则无视(返回None)。

s1.copy()拷贝(浅拷贝)集合s1。

len(s1)返回集合s1长度。

i in s1测试元素i是否在集合s1中。

除了这些基本操作外,还有基于集合运算的修改操作。

取得并集后覆盖集合s1:

s1.update(*others)
s1 |= other |...

取得交集后覆盖集合s1:

s1.intersection_update(*others)
s1 &= other & ...

取得差集后覆盖集合s1:

s1.difference_update(*others)
s1 -= other |...

取得XOR运算后的结果覆盖集合s1:

s1.symmetric_difference_update(other)
s1 ^= other

集合解析

到目前为止,各种解析表达式的方式已经很清晰了。所以看示例即可:

>>> {x for x in 'abcde'}
{'b', 'e', 'c', 'a', 'd'} >>> {c*2 for c in "abcde"}
{'ee', 'bb', 'cc', 'dd', 'aa'}