Python中使用字典的几个小技巧

1 解包

所谓解包,就是将字典通过 ** 操作符转为 Key=Value 的形式,这种形式可以直接传给函数作为关键字参数。
说说适用的几种情况。

1.1 搜索拼接条件

当应用中使用类似 SQLAlchemy 的 ORM 形式读取数据的时候,不同搜索条件,传入给 ORM 的搜索参数也随之改变。
下面是图书表的部分数据(只展示了部分字段)

+----+---------------+-------------------------+-------+
| id | category_name | book_name               | price |
+----+---------------+-------------------------+-------+
|  1 | 人文社科      | 人类简史                | 42.90 |
|  2 | 人文社科      | 世界简史                | 25.50 |
|  3 | 经济管理      | 极致产品                | 37.00 |
|  4 | 经济管理      | 史蒂夫·乔布斯传         | 44.20 |
|  5 | 经济管理      | 影响力                  | 41.20 |
+----+---------------+-------------------------+-------+

搜索时,我们会以这样的形式执行查询方法

books = Book.query.filter_by(id=1, book_name='影响力').all()

但是由于传入参数会根据搜索条件的变化而变化,无法直接写出有哪些参数,这个时候就可以使用字典解包

condition = {}
if book_id:
    condition['id'] = id
if book_name:
    condition['name'] = book_name
books = Book.query.filter_by(**condition).all()

这样就 OK 了

1.2 方法参数太多,为代码美观使用

new_book = Book(category_name='文学小说', book_name='解忧杂货店', price=28.8,
                ...)
db.session.add(new_book)

改成这样的话,美观一些

book_param = {'category_name': '文学小说', 'book_name': '解忧杂货店', 'price': 28.8,
              ...}
new_book = Book(**book_param)
db.session.add(new_book)

并且,在上述新增图书过程中,都会对提交的参数进行校验,而校验方法返回的结果(也就是 book_param 和其它信息)一般也都是字典,所以使用字典解包的方式更符合实际场景。

总之,适当使用字典解包对方法进行传参,可以让我们的代码更灵活。

2 setdefault() 的使用

先看下这个方法怎么使用

dict.setdefault(key, default=None)

如果字典中包含有给定键,则返回该键对应的值,否则返回为该键设置的值。

很多时候我们需要对列表根据元素的某个 key 转化成一个包含列表的字典。比如,上面的数据中,我希望得到一个字典,字典的 key 是图书分类,value 是属于该分类的图书列表。我们通常会这样写

books_dict = {}
for book in book_list:
    if book['category_name'] not in books_dict.keys():
        books_dict[book['category_name']] = []
    books_dict[book['category_name']].append(book)

当然,这样写是正确的,能得到预期结果

{
    "人文社科": [{
        "id": 1,
        "category_name": "人文社科",
        "book_name": "人类简史",
        "price": 42.9
    }, {
        "id": 2,
        "category_name": "人文社科",
        "book_name": "世界简史",
        "price": 25.5
    }],
    "经济管理": [{
        "id": 3,
        "category_name": "经济管理",
        "book_name": "极致产品",
        "price": 37.0
    }, {
        "id": 4,
        "category_name": "经济管理",
        "book_name": "史蒂夫·乔布斯传",
        "price": 44.2
    }, {
        "id": 5,
        "category_name": "经济管理",
        "book_name": "影响力",
        "price": 41.2
    }]
}

但是如果使用字典的 setdefault() 方法话,可以少写几行代码,看起来也优雅一些

books_dict = {}
for book in book_list:
    books_dict.setdefault(book['category_name'], []).append(book)

3 字典合并

常用的合并方式

# new_dict = {**dict1, **dict2, ...}
# 合并多个字典,如果字典中存在相同的 key 的话,后面的会覆盖掉前面的
# 比如 dict2 会覆盖 dict1 中的 key 相同的值

>>> a = {'name': 'x', 'age': 13}
>>> b = {'name': 'y'}
>>> c = {**a, **b}
>>> c
{'name': 'y', 'age': 13}

# dict1.update(dict2)
# 合并两个字典,如果字典中存在相同的 key 的话,dict2 会覆盖 dict1 的对应值
# 理解为更新某个字典应该更合适

>>> a.update(b)
>>> a
{'name': 'y', 'age': 13}

有时我们碰到合并字典的情况也不少。比如,我们准备根据一本书的基本信息创建一本新书

# to_dict 将 ORM 对象转为字典,是自定义的,理解意思就好
base_book = Book.query.filter_by(id=1).first().to_dict()
# 提交的参数需要校验,校验成功后返回值包含 book_param ,内容和下面类似
book_param = {'book_name': '国家宝藏', 'price': 55.60}
# 同时需要更新新书的创建时间和更新时间
time_param = {'created_at': current_time, 'updated_at': current_time}
# 新增书籍
new_book = Book(**{**base_book, **book_param, **time_param})
db.session.add(new_book)

当然,如果只是合并两个字典的话,也可以使用 update() 方法。

假设我们只需要合并 base_bookbook_param

base_book.update(book_param)

这也可以工作,不过要注意,这样会修改 base_book 中的值。

如果只是单纯的更新某个字典的信息的话,update() 方法显然最合适。对于当前需求的话,还是第一种方式更合适。

本文首发于公众号「小小后端」,关注并回复「HMPython2018」获取 2018 最新 Python 学习资源。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 230,622评论 6 544
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 99,716评论 3 429
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 178,746评论 0 383
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 63,991评论 1 318
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 72,706评论 6 413
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 56,036评论 1 329
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 44,029评论 3 450
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 43,203评论 0 290
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 49,725评论 1 336
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 41,451评论 3 361
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 43,677评论 1 374
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 39,161评论 5 365
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 44,857评论 3 351
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 35,266评论 0 28
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 36,606评论 1 295
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 52,407评论 3 400
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 48,643评论 2 380

推荐阅读更多精彩内容