快捷導(dǎo)航

書寫Python代碼的一種更優(yōu)雅方式(推薦!)

更新時間：2022年10月21日 11:47:55 作者：今天代碼沒bug

Python是一個高層次的結(jié)合了解釋性、編譯性、互動性和面向?qū)ο蟮哪_本語言,下面這篇文章主要給大家介紹了關(guān)于書寫Python代碼的一種更優(yōu)雅方式,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下

1 簡介

一些比較熟悉pandas的讀者朋友應(yīng)該經(jīng)常會使用query()、eval()、pipe()、assign()等pandas的常用方法，書寫可讀性很高的「鏈式」數(shù)據(jù)分析處理代碼，從而更加絲滑流暢地組織代碼邏輯。

但在原生Python中并沒有提供類似shell中的管道操作符|、R中的管道操作符%>%等語法，也沒有針對列表等數(shù)組結(jié)構(gòu)的可進行鏈式書寫的快捷方法，譬如javascript中數(shù)組的map()、filter()、some()、every()等。

正所謂“標準庫不夠，三方庫來湊”，Python原生對鏈式寫法支持不到位沒關(guān)系，我們可以使用一些簡單方便且輕量的第三方庫來協(xié)助我們在Python代碼中大面積實現(xiàn)鏈式寫法，今天的文章中我就將帶大家一起學(xué)習(xí)相關(guān)的知識技巧~

2 在Python中配合pipe靈活使用鏈式寫法

我們將使用到pipe這個第三方庫，它不僅內(nèi)置了很多實用的「管道操作函數(shù)」，還提供了將常規(guī)函數(shù)快捷「轉(zhuǎn)換」為管道操作函數(shù)的方法，使用pip install pipe對其進行安裝即可。

pipe的用法非常方便，類似shell中的管道操作：以你的數(shù)組變量為起點，使用操作符|銜接pipe內(nèi)置的各個常見管道操作函數(shù)，組裝起自己所需的計算步驟即可，譬如，我們篩選輸入數(shù)組中為偶數(shù)的，再求平方，就可以寫作：

import pipe
 
list(
    range(10) | 
    pipe.filter(lambda x: x % 2 == 0) | 
    pipe.select(lambda x: x ** 2)
)

因為pipe搭建的管道默認都是惰性運算的，直接產(chǎn)生的結(jié)果是生成器類型，所以上面的例子中我們最外層套上了list()來取得實際計算結(jié)果，更優(yōu)雅的方式是配合pipe.Pipe()，將list()也改造為管道操作函數(shù)：

from pipe import Pipe
 
(
    range(10) | 
    pipe.filter(lambda x: x % 2 == 0) | 
    pipe.select(lambda x: x ** 2) |
    Pipe(list)
)

在上面的簡單例子中我們使用到的filter()、select()等就是pipe中常見的管道操作函數(shù)，事實上pipe中的管道操作函數(shù)相當?shù)呢S富，下面我們來展示其中一些常用的：

2.1 pipe中常用的管道操作函數(shù)

2.1.1 使用traverse()展平嵌套數(shù)組

如果你想要將任意嵌套數(shù)組結(jié)構(gòu)展平，可以使用traverse()：

(
    [1, [2, 3, [4, 5]], 6, [7, 8, [9, [10, 11]]]] | 
    pipe.traverse | 
    Pipe(list)
)

2.1.2 使用dedup()進行順序去重

如果我們需要對包含若干重復(fù)值的數(shù)組進行去重，且希望保留原始數(shù)據(jù)的順序，則可以使用dedup()，其還支持key參數(shù)，類似sorted()中的同名參數(shù)，實現(xiàn)自定義去重規(guī)則：

(
    [-1, 0, 0, 0, 1, 2, 3] | 
    pipe.dedup |
    Pipe(list)
)
 
(
    [-1, 0, 0, 0, 1, 2, 3] | 
    # 基于每個元素的絕對值進行去重
    pipe.dedup(key=abs) |
    Pipe(list)
)

2.1.3 使用filter()進行值過濾

我們最開始的例子中使用過它，用法就是基于傳入的lambda函數(shù)對每個元素進行條件判斷，并保留結(jié)果為True的，與javascript中的filter()方法非常相似：

(
    [1, 4, 3, 2, 5, 6, 8] |
    # 保留大于5的元素
    pipe.filter(lambda x: x > 5) |
    Pipe(list)
)

2.1.4 使用groupby()進行分組運算

這個函數(shù)非常實用，其功能相當于管道操作版本的itertools.groupby()，可以幫助我們基于lambda函數(shù)運算結(jié)果對原始輸入數(shù)組進行分組，通過groupby()操作后直接得到的結(jié)果是分組結(jié)果的二元組列表，每個元組的第一個元素是分組標簽，第二個元素是分到該組內(nèi)的各個元素：