刚刚学python的童鞋们在学习set集合的时候,肯定会有一个疑问。它有两个删除的方法,到底该用哪个呢?那你得知道,程序不会平白无故的将达到效果的方法写两遍的,存在肯定有它存在的意义。
那么这次就剖析一下两个删除元素方法:discard和remove的区别在哪里。
先来看看set集合。简单说,set就是一个不允许重复元素的集合,跟数学里的集合概念很像。它里面的元素是无序的,所以不能用索引来访问特定位置的元素。
先认识一下remove方法
点到remove方法里面去看一下
remove()方法用起来很简单:你想删除哪个元素,就直接把它作为参数传进去。比如:
python
my_set = {"apple", "banana", "cherry"}
my_set.remove("banana")
print(my_set) # 输出:{'apple', 'cherry'}
点到里面去看的话,把那个英文翻译出来就是:从集合中移除一个元素;他必须是成员,如果元素不是成员,则引发keyError错误
那我们就试试它这个错误(看下图:果真如上面的所说,报错了)

再来看看discard方法
discard()方法跟remove()干的事情差不多,也是删除指定元素,同样的我们先看看这个方法

看起来跟remove()效果一样对吧?区别在于:当你试图删除一个不存在的元素时,discard()不会报错,它会默默地什么都不做,程序继续正常运行。

什么时候用哪个?
现在你明白区别了,那么实际编程中该怎么选择呢?
如果你十分确定要删除的元素肯定存在于集合中,而且如果不存在就是程序出了错,那么用remove()更合适。因为一旦元素不存在,它会立即抛出异常提醒你有问题需要处理。
如果你不确定元素是否存在,或者元素不存在也没关系,你只是想确保它不在集合中,那么用discard()更合适。它不会因为元素不存在而打断程序的执行。
举个例子,假设你在处理用户输入的兴趣标签:
python
# 用户选择的标签
user_tags = {"音乐", "电影", "读书"}
# 用户移除了"电影"标签
if "电影" in user_tags:
user_tags.remove("电影")
# 或者更简单的写法:
user_tags.discard("电影")
第二种写法更简洁吧?因为discard()自己会先检查元素是否存在。
还有一个pop()方法
顺便提一下,set还有个pop()方法,它不需要参数,会随机删除并返回一个元素(因为set是无序的)。如果set是空的,它也会抛出KeyError。
python
my_set = {"apple", "banana", "cherry"}
element = my_set.pop()
print(f"删除了: {element}, 剩下的: {my_set}")
总结一下
remove()和discard()都是set集合中删除元素的方法,它们的关键区别在于处理不存在元素时的行为:
- remove():元素必须存在,不存在就报错(KeyError)
- discard():元素存在就删除,不存在也不报错
选择哪个取决于你的需求:如果你希望元素不存在时得到错误提示,用remove();如果你不在乎元素是否存在,只是希望确保它不在集合中,用discard()。
简单记就是:remove()更严格,discard()更随意。
好了,关于set集合删除元素的事儿就聊到这里。希望这次讲解能帮你理清这两个方法的区别,下次用的时候就不会迷糊了!