Python 集合 remove() 方法

---

---

Python 集合的 remove() 方法用于从集合中删除指定的元素。如果元素存在于集合中，则将其删除。如果找不到元素，则会引发 KeyError 异常。

还有 discard() 方法，它也删除元素，但如果元素不存在则不会引发错误，而 remove() 方法确保指定的元素必须存在于集合中。

当我们需要在删除元素之前强制元素存在时，此方法很有用，它通过显式处理意外缺失的元素来确保集合的完整性。

语法

以下是 Python 集合 remove() 方法的语法和参数。

set.remove(element)

参数

此方法仅接受一个参数，即 element，如果存在，则为要从集合中删除的元素。

返回值

此方法不返回任何值。

示例 1

以下示例演示了如何删除存在于集合中的元素：

# Define a set
my_set = {1, 2, 3, 4, 5}

# Remove an element
my_set.remove(3)

print(my_set)  

输出

{1, 2, 4, 5}

示例 2

此示例尝试删除集合中不存在的元素，并引发 KeyError：

# Define a set
my_set = {1, 2, 3}

# Try to remove a non-existing element
try:
    my_set.remove(4)
except KeyError as e:
    print(f"Error: {e}")  

输出

Error: 4

示例 3

现在，在这里我们使用循环从集合中删除多个元素。

# Define a set
my_set = {1, 2, 3, 4, 5}

# Remove multiple elements
elements_to_remove = {2, 4}
for elem in elements_to_remove:
    my_set.remove(elem)

print(my_set)  

输出

{1, 3, 5}

示例 4

在此示例中，我们使用列表推导式并在列表推导式中定义条件以根据该条件删除元素：

# Define a set
my_set = {1, 2, 3, 4, 5}

# Remove elements greater than 3
my_set = {elem for elem in my_set if elem <= 3}

print(my_set) 

输出

{1, 2, 3}