python glom模塊的使用簡介

更新時間：2021年04月12日 15:16:53 作者：派森醬

這篇文章主要介紹了python glom模塊的使用簡介，幫助大家更好的利用python處理數(shù)據(jù)，感興趣的朋友可以了解下

安裝

作為 Python 內(nèi)置模塊，相信你一定知道怎么安裝：

pip3 install glom

幾秒鐘就搞定！

簡單使用

我們來看看最簡單的用法：

d = {"a": {"b": {"c": 1}}}
print(glom(d, "a.b.c")) # 1

在這里，我們有一個嵌套三層的 json 結(jié)構(gòu)，我們想獲取最里層的 c 對應(yīng)的值，正常的寫法應(yīng)該是：

print(d["a"]["b"]["c"])

如果到這里，我說 glom 比傳統(tǒng)方式好一些，因為你不用一層層地寫中括號和引號，你會不會嗤之以鼻？

好，我們再來看看下面的情況：

d = {"a": {"b": None}}
print(d["a"]["b"]["c"])

遍歷到一個 None 對象，你會收到下面的錯誤：

Traceback (most recent call last):
  File "/Users/cxhuan/Documents/python_workspace/mypy/pmodules/pglom/glomstudy.py", line 10, in <module>
    print(d["a"]["b"]["c"])
TypeError: 'NoneType' object is not subscriptable

我們來看看 glom 的處理方式：

from glom import glom

d = {"a": {"b": None}}
print(glom(d, "a.b.c"))

同樣地，glom 不能把錯誤的輸出成對的，你會得到以下錯誤：

Traceback (most recent call last):
  File "/Users/cxhuan/Documents/python_workspace/mypy/pmodules/pglom/glomstudy.py", line 11, in <module>
    print(glom(d, "a.b.c"))
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/glom/core.py", line 2181, in glom
    raise err
glom.core.PathAccessError: error raised while processing, details below.
 Target-spec trace (most recent last):
 - Target: {'a': {'b': None}}
 - Spec: 'a.b.c'
glom.core.PathAccessError: could not access 'c', part 2 of Path('a', 'b', 'c'), got error: AttributeError("'NoneType' object has no attribute 'c'")

如果你仔細(xì)看報錯內(nèi)容，你就會發(fā)現(xiàn)這報錯內(nèi)容極其詳細(xì)，一目了然，這對于找程序 bug 簡直是神器！

復(fù)雜用法

剛才簡單的例子，讓大家對 glom 有了直觀的認(rèn)識，接下來我們看看 glom 的 glom 方法的定義：

glom(target, spec, **kwargs)

我們看看參數(shù)的含義：

target：目標(biāo)數(shù)據(jù)，可以是dict、list或者其他任何對象
spec：是我們希望輸出的內(nèi)容

下面我們來使用這個方法。

先看一個例子。我們有一個 dict ，想要獲取出所有 name 的值，我們可以通過 glom 來實現(xiàn)：

data = {"student": {"info": [{"name": "張三"}, {"name": "李四"}]}}
info = glom(data, ("student.info", ["name"]))
print(info) # ['張三', '李四']

如果用傳統(tǒng)方式的話，我們可能會需要遍歷才能獲取到，但是使用 glom ，我們只需要一行代碼就可以了，輸出是一個數(shù)組。

如果你不想輸出數(shù)組，而是想要一個 dict 的話，那也是很簡單的：

info = glom(data, {"info": ("student.info", ["name"])})
print(info) # {'info': ['張三', '李四']

我們只需要將原來的數(shù)組賦值給一個字典來接收就好了。

搞定麻煩需求

假如我現(xiàn)在有兩組數(shù)據(jù)，我要取出 name 的值：

data_1 = {"school": {"student": [{"name": "張三"}, {"name": "李四"}]}}
data_2 = {"school": {"teacher": [{"name": "王老師"}, {"name": "趙老師"}]}}

spec_1 = {"name": ("school.student", ["name"])}
spec_2 = {"name": ("school.teacher", ["name"])}
print(glom(data_1, spec_1)) # {'name': ['張三', '李四']}
print(glom(data_2, spec_2)) # {'name': ['王老師', '趙老師']}

我們通常是這么寫，對嗎？假如我們有好多組數(shù)據(jù)，每組都是類似的取法呢？這時候我們就會想辦法避免一個個重復(fù)寫 N 行參數(shù)了，我們可以使用 Coalesce 方法：

data_1 = {"school": {"student": [{"name": "張三"}, {"name": "李四"}]}}
data_2 = {"school": {"teacher": [{"name": "王老師"}, {"name": "趙老師"}]}}

spec = {"name": (Coalesce("school.student", "school.teacher"), ["name"])}
 
print(glom(data_1, spec)) # {'name': ['張三', '李四']}
print(glom(data_2, spec)) # {'name': ['王老師', '趙老師']}

我們可以用 Coalesce 把多個需求聚合起來，然后針對同一個 spec 來取值就行了。

下面再來一個大殺器——取值計算。glom 還可以對取值進(jìn)行簡單計算，我們來看例子：

data = {"school": {"student": [{"name": "張三", "age": 8}, {"name": "李四", "age": 10}]}}
spec = {"sum_age": ("school.student", ["age"], sum)}
print(glom(data, spec)) # {'sum_age': 18}