pythonで集合setに要素を追加・削除する方法

プログラミング python

概要

pythonのsetにおいて要素を追加・削除する方法を説明する。

集合に要素を追加

add()で行う

s = {1, 2, 3}
s.add(4)
print(s)
# {1, 2, 3, 4}

集合の要素を削除

集合の要素を削除するメソッドは、 discard(), remove(), pop(), clear()の4つがある。

  • discard()について
    引数に指定した要素を削除。集合に存在しない要素を指定した場合は何もしない。
s = {1, 2, 3}
s.discard(2)
print(s)
# {1, 3}
s.discard(100)
print(s)
# {1, 3}
  • remove()について
    引数に指定した要素を削除。集合に存在しない要素を指定した場合はKey Errorとなる。
s = {1, 2, 3}
s.remove(2)
print(s)
# {1, 3}
s.remove(100)
# Traceback (most recent call last):
# File "<stdin>", line 1, in <module>
# KeyError: 100
  • pop()について
    集合から要素を削除する。どの要素が削除されるかは選べない。空集合に対して実施した場合はKey Errorとなる。
s = {1, 2, 3}
v = s.pop()
print(v)
# 1
print(s)
# {2, 3}

s = {1}
s.pop()
# 1
s.pop()
# Traceback (most recent call last):
# File "<stdin>", line 1, in <module>
# KeyError: 'pop from an empty set'
  • clear()について
    clear()メソッドは集合のすべての要素を削除し、空集合にする。
s = {1, 2, 3}
s.clear()
print(s)
# set()