美文网首页程序员Spring-BootJava学习笔记
Python 关于字典的操作,看这篇文章就够了

Python 关于字典的操作,看这篇文章就够了

作者: 程序员知识圈 | 来源:发表于2022-05-13 14:51 被阅读0次

    字典是 Python 必用且常用的数据结构,本文梳理常用的字典操作,看这个就够了,涉及:

    • 初始化
    • 合并字典
    • 字典推导式
    • Collections 标准库
    • 字典转 JSON
    • 字典转 Pandas

    初始化

    # 最常用这种
    my_object = {
      "a": 5,
      "b": 6
    }
    # 如果你不喜欢写大括号和双引号:
    my_object = dict(a=5, b=6)
    

    合并字典

    a = { "a": 5, "b": 5 }
    b = { "c": 5, "d": 5 }
    c = { **a, **b } #最简单的方式
    assert c == { "a": 5, "b": 5, "c": 5, "d": 5 }
    
    # 合并后还要修改,可以这样:
    c = { **a, **b, "a": 10 }
    assert c == { "a": 10, "b": 5, "c": 5, "d": 5 }
    b["a"] = 10
    c = { **a, **b }
    assert c == { "a": 10, "b": 5, "c": 5, "d": 5 }
    

    字典推导式

    # 使用字典推导式来删除 key
    a = dict(a=5, b=6, c=7, d=8)
    remove = set(["c", "d"])
    a = { k: v for k,v in a.items() if k not in remove }
    # a = { "a": 5, "b": 6 }
    
    # 使用字典推导式来保留 key
    a = dict(a=5, b=6, c=7, d=8)
    keep = remove
    a = { k: v for k,v in a.items() if k in keep }
    # a = { "c": 7, "d": 8 }
    
    # 使用字典推导式来让所有的 value 加 1
    a = dict(a=5, b=6, c=7, d=8)
    a = { k: v+1 for k,v in a.items() }
    # a = { "a": 6, "b": 7, "c": 8, "d": 9 }
    

    Collections 标准库

    Collections 是 Python 中的一个内置模块,它有几个有用的字典子类,可以大大简化 Python 代码。我经常使用的其中两个类,defaultdict 和 Counter。此外,由于它是 dict 的子类,因此它具有标准方法,如 items()、keys()、values() 等。

    from collections import Counter
    
    counter = Counter()
    #counter 可以统计 list 里面元素的频率
    counter.update(['a','b','a']
    #此时 counter = Counter({'a': 2, 'b': 1})
    
    #合并计数
    counter.update({ "a": 10000, "b": 1 })
    # Counter({'a': 10002, 'b': 2})
    counter["b"] += 100
    # Counter({'a': 10002, 'b': 102})
    
    print(counter.most_common())
    #[('a', 10002), ('b', 102)]
    print(counter.most_common(1)[0][0])
    # => a
    

    defaultdict 也是 dict 的必杀技:

    from collections import defaultdict
    
    # 如果字典的 value 是 字典
    a = defaultdict(dict)
    assert a[5] == {}
    a[5]["a"] = 5
    assert a[5] == { "a": 5 }
    
    # 如果字典的 value 是列表
    a = defaultdict(list)
    assert a[5] == []
    a[5].append(3)
    assert a[5] == [3]
    
    # 字典的 value 的默认值可以是 lambda 表达式
    a = defaultdict(lambda: 10)
    assert a[5] == 10
    assert a[6] + 1 == 11
    
    # 字典里面又是一个字典,不用这个,你要做多少初始化操作?
    a = defaultdict(lambda: defaultdict(dict))
    assert a[5][5] == {}
    

    字典转 JSON

    我们通常说的 JSON 就是指 JSON 字符串,它是一个字符串。Dict 可以转成 JSON 格式的字符串。

    import json
    
    a = dict(a=5, b=6)
    
    # 字典转 JSON 字符串
    json_string = json.dumps(a)
    # json_string = '{"a": 5, "b": 6}'
    
    # JSON 字符串转字典
    assert a == json.loads(json_string)
    
    # 字典转 JSON 字符串保存在文件里
    with open("dict.json", "w+") as f:
        json.dump(a, f)
    
    # 从 JSON 文件里恢复字典
    with open("dict.json", "r") as f:
        assert a == json.load(f)
    

    字典转 Pandas

    import pandas as pd
    
    # 字典转 pd.DataFrame
    df = pd.DataFrame([
        { "a": 5, "b": 6 },
        { "a": 6, "b": 7 }
    ])
    # df =
    #    a  b
    # 0  5  6
    # 1  6  7
    
    # DataFrame 转回字典
    a = df.to_dict(orient="records")
    # a = [
    #    { "a": 5, "b": 6 },
    #    { "a": 6, "b": 7 }
    # ]
    
    # 字典转 pd.Series
    srs = pd.Series({ "a": 5, "b": 6 })
    # srs =
    # a    5
    # b    6
    # dtype: int64
    
    # pd.Series 转回字典
    a = srs.to_dict()
    # a = {'a': 5, 'b': 6}
    

    最后的话

    以上就是字典的常用方式,希望对你有所帮助,欢迎点赞、转发、收藏。

    相关文章

      网友评论

        本文标题:Python 关于字典的操作,看这篇文章就够了

        本文链接:https://www.haomeiwen.com/subject/nmtuurtx.html