介紹

本文主要介紹Python中set的基本知識(shí)和使用。

Python中什么是set

dict的作用是建立一組 key 和一組 value 的映射關(guān)系，dict的key是不能重復(fù)的。

有的時(shí)候，我們只想要 dict 的 key，不關(guān)心 key 對(duì)應(yīng)的 value，目的就是保證這個(gè)集合的元素不會(huì)重復(fù)，這時(shí)，set就派上用場(chǎng)了。

set 持有一系列元素，這一點(diǎn)和list很像，但是set的元素沒(méi)有重復(fù)，而且是無(wú)序的，這點(diǎn)和 dict 的 key很像。

創(chuàng)建 set 的方式是調(diào)用 set() 并傳入一個(gè) list，list的元素將作為set的元素：

>>> s = set(['A', 'B', 'C'])

可以查看 set 的內(nèi)容：

>>> print s
set(['A', 'C', 'B'])

請(qǐng)注意，上述打印的形式類(lèi)似 list， 但它不是list，仔細(xì)看還可以發(fā)現(xiàn)，打印的順序和原始 list 的順序有可能是不同的，因?yàn)閟et內(nèi)部存儲(chǔ)的元素是無(wú)序的。

因?yàn)閟et不能包含重復(fù)的元素，所以，當(dāng)我們傳入包含重復(fù)元素的 list 會(huì)怎么樣呢？

>>> s = set(['A', 'B', 'C', 'C'])
>>> print s
set(['A', 'C', 'B'])
>>> len(s)
3

結(jié)果顯示，set會(huì)自動(dòng)去掉重復(fù)的元素，原來(lái)的list有4個(gè)元素，但set只有3個(gè)元素。

Python之 訪問(wèn)set

由于set存儲(chǔ)的是無(wú)序集合，所以我們沒(méi)法通過(guò)索引來(lái)訪問(wèn)。
訪問(wèn) set中的某個(gè)元素實(shí)際上就是判斷一個(gè)元素是否在set中。
例如，存儲(chǔ)了班里同學(xué)名字的set：

>>> s = set(['Adam', 'Lisa', 'Bart', 'Paul'])

我們可以用 in 操作符判斷：
Bart是該班的同學(xué)嗎？

>>> 'Bart' in s
True

Bill是該班的同學(xué)嗎？

>>> 'Bill' in s
False

bart是該班的同學(xué)嗎？

>>> 'bart' in s
False

看來(lái)大小寫(xiě)很重要，'Bart' 和 'bart'被認(rèn)為是兩個(gè)不同的元素。

Python之 set的特點(diǎn)

set的內(nèi)部結(jié)構(gòu)和dict很像，唯一區(qū)別是不存儲(chǔ)value，因此，判斷一個(gè)元素是否在set中速度很快。

set存儲(chǔ)的元素和dict的key類(lèi)似，必須是不變對(duì)象，因此，任何可變對(duì)象是不能放入set中的。

最后，set存儲(chǔ)的元素也是沒(méi)有順序的。

set的這些特點(diǎn)，可以應(yīng)用在哪些地方呢？

星期一到星期日可以用字符串'MON', 'TUE', ... 'SUN'表示。

假設(shè)我們讓用戶(hù)輸入星期一至星期日的某天，如何判斷用戶(hù)的輸入是否是一個(gè)有效的星期呢？

可以用 if 語(yǔ)句判斷，但這樣做非常繁瑣：

x = '???' # 用戶(hù)輸入的字符串
if x!= 'MON' and x!= 'TUE' and x!= 'WED' ... and x!= 'SUN':
  print 'input error'
else:
  print 'input ok'

注意：if 語(yǔ)句中的...表示沒(méi)有列出的其它星期名稱(chēng)，測(cè)試時(shí)，請(qǐng)輸入完整。

如果事先創(chuàng)建好一個(gè)set，包含'MON' ~ 'SUN'：

weekdays = set(['MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT', 'SUN'])

再判斷輸入是否有效，只需要判斷該字符串是否在set中：

x = '???' # 用戶(hù)輸入的字符串
if x in weekdays:
  print 'input ok'
else:
  print 'input error'

這樣一來(lái)，代碼就簡(jiǎn)單多了。

Python之 遍歷set

由于 set 也是一個(gè)集合，所以，遍歷 set 和遍歷 list 類(lèi)似，都可以通過(guò) for 循環(huán)實(shí)現(xiàn)。
直接使用 for 循環(huán)可以遍歷 set 的元素：

>>> s = set(['Adam', 'Lisa', 'Bart'])
>>> for name in s:
...   print name
... 
Lisa
Adam
Bart

注意: 觀察 for 循環(huán)在遍歷set時(shí)，元素的順序和list的順序很可能是不同的，而且不同的機(jī)器上運(yùn)行的結(jié)果也可能不同。

Python之 更新set

由于set存儲(chǔ)的是一組不重復(fù)的無(wú)序元素，因此，更新set主要做兩件事：
一是把新的元素添加到set中，二是把已有元素從set中刪除。

添加元素時(shí)，用set的add()方法：

>>> s = set([1, 2, 3])
>>> s.add(4)
>>> print s
set([1, 2, 3, 4])

如果添加的元素已經(jīng)存在于set中，add()不會(huì)報(bào)錯(cuò)，但是不會(huì)加進(jìn)去了：

>>> s = set([1, 2, 3])
>>> s.add(3)
>>> print s
set([1, 2, 3])

刪除set中的元素時(shí)，用set的remove()方法：

>>> s = set([1, 2, 3, 4])
>>> s.remove(4)
>>> print s
set([1, 2, 3])

如果刪除的元素不存在set中，remove()會(huì)報(bào)錯(cuò)：

>>> s = set([1, 2, 3])
>>> s.remove(4)
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
KeyError: 4

所以用add()可以直接添加，而remove()前需要判斷。

結(jié)語(yǔ)

以上就是python開(kāi)發(fā)入門(mén)——set的使用的詳細(xì)內(nèi)容，更多關(guān)于python set的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論