作者：豆豆

來(lái)源：Python 技術(shù)

眾所周知，Pytnon 非常擅長(zhǎng)處理數(shù)據(jù)，尤其是后期數(shù)據(jù)的清洗工作。今天派森醬就給大家介紹一款處理數(shù)據(jù)的神器 Pipe。

什么是 Pipe

簡(jiǎn)言之，Pipe 是 Python 的一個(gè)三方庫(kù)。

通過(guò) Pipe 我們可以將一個(gè)函數(shù)的處理結(jié)果傳遞給另外一個(gè)函數(shù)，這意味著你的代碼會(huì)非常簡(jiǎn)潔。

要使用 Pipe 需要提前安裝，直接使用 pip 安裝即可。

pip install pipe 

過(guò)濾元素

和 filter 類似，pipe 中的 where 操作可以過(guò)濾可迭代對(duì)象中的元素。

In [5]: numbers = [0, 1, 2, 3, 4, 5] In [6]: list(numbers | where(lambda x: x % 2 == 0)) Out[6]: [0, 2, 4] 

作用元素

類似 map，select 操作可以將函數(shù)作用于可迭代對(duì)象中的每個(gè)元素。下面的例子中我們將列表中的元素都擴(kuò)大 2 倍。

In [8]: list(numbers | select(lambda x: x * 2)) Out[8]: [0, 2, 4, 6, 8, 10] 

當(dāng)然，還可以將多種操作合并在一起來(lái)玩。

下面的例子就是將列表中的偶數(shù)挑選出來(lái)并擴(kuò)大 2 倍，和 filter 與 map 不同的是，pipe 可以將多個(gè)操作連接起來(lái)，就像水管套水管一樣，所以我想管道這個(gè)名字也是很接地氣了。

In [10]: list(numbers ...: | where(lambda x: x % 2 == 0) ...: | select(lambda x: x * 2) ...: ) ...: Out[10]: [0, 4, 8] 

連接元素

操作嵌套列表時(shí)非常痛苦，值得高興的是 pipe 給出了很友好的接口，只需要 chain 一下即可。

In [11]: list([[1, 2], [3, 4], [5]] | chain) Out[11]: [1, 2, 3, 4, 5] In [30]: list((1, 2, 3) | chain_with([4, 5], [6])) Out[30]: [1, 2, 3, 4, 5, 6] In [31]: list((1, 2, 3) | chain_with([4, 5], [6,[7]])) Out[31]: [1, 2, 3, 4, 5, 6, [7]] 

如你所見(jiàn)，chain 只可以拆開(kāi)一層，如果要拆開(kāi)多層嵌套的話，不要慌，traverse 輕松搞定。

In [12]: list([[1, 2], [[[3], [[4]]], [5]]] | traverse) Out[12]: [1, 2, 3, 4, 5] 

結(jié)合 select 一起，獲取字典中的某個(gè)字段屬性集合。

In [32]: fruits = [
    ...:     {"name": "apple", "price": [2, 5]},
    ...:     {"name": "orange", "price": 4},
    ...:     {"name": "grape", "price": 5},
    ...: ]

In [33]: list(fruits
    ...:      | select(lambda fruit: fruit["price"])
    ...:      | traverse)
    ...:
Out[33]: [2, 5, 4, 5]

分組

對(duì)列表中的元素進(jìn)行分組是必不可少的，在 pipe 中可以使用 groupby 來(lái)完成。

In [26]: list(numbers ...: | groupby(lambda x: 'Even' if x % 2 == 0 else 'Odd') ...: | select(lambda x: {x[0]: list(x[1])}) ...: ) ...: Out[26]: [{'Even': [0, 2, 4]}, {'Odd': [1, 3, 5]}] 

同樣，還可以在 select 中添加 where 過(guò)濾條件。

In [27]: list(numbers
    ...:      | groupby(lambda x: 'Even' if x % 2 == 0 else 'Odd')
    ...:      | select(lambda x: {x[0]: list(x[1] | where(lambda x: x > 2))})
    ...:     )
    ...:
Out[27]: [{'Even': [4]}, {'Odd': [3, 5]}]

行列互換

數(shù)據(jù)處理中時(shí)常會(huì)用到行列互相轉(zhuǎn)換，尤其是在用 DataFrame 時(shí)，使用 pipe 一行代碼搞定行列轉(zhuǎn)換。

In [24]: [[1, 2, 3], [4, 5, 6], [7, 8, 9]] | transpose Out[24]: [(1, 4, 7), (2, 5, 8), (3, 6, 9)] 

刪除元素

對(duì)列表去重也是一項(xiàng)常用的操作，在 pipe 中使用 dedup 來(lái)對(duì)列表進(jìn)行去重。

In [28]: list([1, 1, 2, 2, 3, 3, 1, 2, 3] | dedup) Out[28]: [1, 2, 3] 

與 dedup 不同的是，uniq 只會(huì)對(duì)連續(xù)的重復(fù)元素保留一個(gè)，非連續(xù)重復(fù)元素則不過(guò)濾。

In [29]: list([1, 1, 2, 2, 3, 3, 1, 2, 3] | uniq) Out[29]: [1, 2, 3, 1, 2, 3] 

總結(jié)

今天派森醬給大家介紹了一個(gè)處理數(shù)據(jù)的神器，使用管道可以讓繁瑣的操作濃縮在幾行甚至一行代碼搞定，提高可讀性的同時(shí)還提升了代碼的整潔程度，美滋滋～