Python 中的字典get()方法和 setdefault()

liftword5个月前 (01-13)技术文章45

Python 字典是强大的数据结构,允许使用键值对有效地存储和检索数据。字典提供的最有用的方法之一是 get() 方法。此方法有助于正常处理缺失的 key,避免常见的 KeyError 异常。此外,setdefault() 方法可用于简化键默认值的管理。

使用get()方法

get() 方法用于从字典中检索与指定关联的值。如果键不存在,则返回默认值(如果未指定,则为 None)。这有助于避免 KeyError 异常,并使代码更加健壮。

value = dictionary.get(key, default_value)
  • key:在字典中搜索的 key。
  • default_value:找不到 key 时返回的值。如果未指定,则默认为 None

get()方法的示例

my_dict = {'a': 1, 'b': 2, 'c': 3}

# Access existing key
value = my_dict.get('b')
print(value)  # Output: 2

# Access non-existing key without default value
value = my_dict.get('d')
print(value)  # Output: None

# Access non-existing key with default value
value = my_dict.get('d', 4)
print(value)  # Output: 4

在此示例中,当在 my_dict 中找不到键 'd' 时,get() 方法默认返回 None,如果指定为默认值,则返回 4

优雅地处理丢失的密钥

使用 get() 方法有助于在处理可能缺少键的字典时避免异常。这在处理从外部源(如 JSON 响应或配置文件)解析的数据时特别有用。

config = {'host': 'localhost', 'port': 8080}

# Retrieve existing key
host = config.get('host')
print(host)  # Output: 'localhost'

# Retrieve non-existing key with default value
timeout = config.get('timeout', 30)
print(timeout)  # Output: 30

这得到了 30 的超时。

使用setdefault()方法

setdefault() 方法类似于 get(),但如果 key 不存在,它还会为键设置默认值。当您想要确保键在字典中具有默认值时,这可能很有用。

value = dictionary.setdefault(key, default_value)

setdefault()方法的示例

my_dict = {'a': 1, 'b': 2}

# Access existing key
value = my_dict.setdefault('b', 0)
print(value)  # Output: 2
print(my_dict)  # Output: {'a': 1, 'b': 2}

# Access non-existing key and set default value
value = my_dict.setdefault('c', 3)
print(value)  # Output: 3
print(my_dict)  # Output: {'a': 1, 'b': 2, 'c': 3}

在此示例中,setdefault() 方法检查 my_dict 中是否存在键 'c'。由于它没有,因此它会添加默认值为 3 的键。

简化数据初始化

setdefault() 方法在初始化嵌套字典或字典中的列表时特别有用。

data = {}

# Initialize nested dictionary if not present
data.setdefault('user', {}).setdefault('preferences', {}).setdefault('theme', 'light')

print(data)  # Output: {'user': {'preferences': {'theme': 'light'}}}

在此示例中,setdefault() 用于确保创建嵌套字典结构(如果尚不存在),并根据需要设置默认值。

字典的 get() 方法是优雅地处理缺失键的重要工具,提供了一种返回默认值而不是引发异常的方法。setdefault() 方法通过在缺少键时设置默认值来进一步简化字典管理。这些方法共同使您的代码更加健壮且更易于维护,尤其是在处理复杂的数据结构或来自外部源的数据时。

相关文章

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json 简介:Json,全名 JavaScript Object Notation,JSON(JavaScript Object Notation(记号、标记))是一种轻量级的数据交换格式。它基于J...

那些让人直呼卧槽的Python代码

作者:闲欢来源:Python 技术Python 是一个设计优美的解释型高级语言, 它提供了很多能让程序员感到舒适的功能特性。但有的时候, Python 的一些输出结果对于初学者来说似乎并不是那么一目了...

Python - 爬虫之Selenium

一、Selenium 的介绍Selenium 是一个 Web 自动化测试工具,最初是为网站自动化测试而开发,Selenium 可以直接调用浏览器,它支持所有主流的浏览器(包括 PhantomJS 这些...