Python 中的一些小技巧

這里是本人收集的一些 Python 小技巧,目前主要是一些實(shí)用函數(shù)业踢,適合有一定基礎(chǔ)的童鞋觀看(不會(huì)專門介紹使用到的標(biāo)準(zhǔn)庫(kù)函數(shù))。秘蛔。

一陨亡、函數(shù)式編程

函數(shù)式編程用來(lái)處理數(shù)據(jù),感覺(jué)很方便深员。(要是再配上管道操作符 | 或者 Java 的那種鏈?zhǔn)秸{(diào)用负蠕,超級(jí)爽!可惜 Python 都沒(méi)有倦畅。遮糖。需要借助第三方庫(kù))

1. 分組/group

數(shù)據(jù)處理中一個(gè)常見(jiàn)的操作,是將列表中的元素叠赐,依次每 k 個(gè)分作一組欲账。

def group_each(a, size: int):
    """
        將一個(gè)可迭代對(duì)象 a 內(nèi)的元素, 每 size 個(gè)分為一組
        group_each([1,2,3,4], 2) -> [(1,2), (3,4)]
    """
    iterators = [iter(a)] * size  # 將新構(gòu)造的 iterator 復(fù)制 size 次(淺復(fù)制)
    return zip(*iterators)  # 然后 zip

這個(gè)函數(shù)之前在 Python 拾遺 - 奇技淫巧 中就寫過(guò)屡江,記得是某次 Google 時(shí)在 stackoverflow 上發(fā)現(xiàn)的,不過(guò)它的最初來(lái)源應(yīng)該是 Python 官方文檔的某個(gè)角落赛不。

順便如果某個(gè) size 比較常用(比如 2)惩嘉,還可以用 partial 封裝一下

from functools import partial

 # 每?jī)蓚€(gè)分一組
group_each_2 = partial(group_each, size=2)  # 等同于 group_each_2 = lambda a: group_each(a, 2)

2. 扁平版本的 map

稍微接觸過(guò)函數(shù)式應(yīng)該都知道 flat_map,可 Python 標(biāo)準(zhǔn)庫(kù)卻沒(méi)有提供踢故。下面是我在 stackoverflow 上找到的實(shí)現(xiàn)文黎,其實(shí)很簡(jiǎn)單

from itertools import chain

def flat_map(f, items):
    return chain.from_iterable(map(f, items))

它和 map 的差別在于是不是扁平(flat) 的(廢話。殿较。)耸峭,舉個(gè)例子

>>> list(map(list, ['123', '456']))
[['1', '2', '3'], ['4', '5', '6']]
>>> list(flat_map(list, ['123', '456']))
['1', '2', '3', '4', '5', '6']

3. 上述函數(shù)的應(yīng)用舉例

在做爬蟲工作時(shí),有時(shí)會(huì)遇到這樣的 table 元素:


image

對(duì)這種 html 元素淋纲,我一般會(huì)直接把它轉(zhuǎn)換成 list劳闹,結(jié)果如下:

table = [['label1', 'value1', 'label2', 'value2'],
         ['label3', 'value3'],
         ['label4', 'value4', 'label5', 'value5'],
         ...
         ]

為了方便索引,現(xiàn)在我需要把上面的數(shù)據(jù)轉(zhuǎn)換成下面這個(gè)樣子的 dict

{
    'label1': 'value1',
    'label2': 'value2',
    'label3': 'value3',
    'label4': 'value4',
    'label5': 'value5'
}

如果是平常洽瞬,大概需要寫循環(huán)了本涕。不過(guò)如果用剛剛說(shuō)到的幾個(gè)函數(shù)的話,會(huì)變得異常簡(jiǎn)單

 # 1\. 分組
groups = flat_map(group_each_2, table)

# 1.1 flat_map 返回的是迭代器片任,list 后內(nèi)容如下:
# [('label1', 'value1'),
#  ('label2', 'value2'),
#  ('label3', 'value3'),
#  ('label4', 'value4'),
#  ('label5', 'value5')]

# 2\. 轉(zhuǎn)換成 dict
key_values = dict(groups)   # 得到的 key_values 與上面需要的 dict 別無(wú)二致偏友。

二蔬胯、其他

1. 多 dict 的去重

假設(shè)我們有一個(gè) dict 的列表对供,里面可能有內(nèi)容一模一樣的 dict,我們需要對(duì)它做去重氛濒。
容易想到的方法就是使用 set产场,可是 set 中的元素必須是 hashable 的,而 dict 是 unhashable 的舞竿,因此不能直接放進(jìn) set 里京景。

>>> a = [{'a': 1}, {'a': 1}, {'b': 2}]
>>> set(a)
Traceback (most recent call last):
  File "/usr/local/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 2961, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-5-5b4c643a6feb>", line 1, in <module>
    set(a)
TypeError: unhashable type: 'dict'

難道就必須手寫遞歸了么?未必骗奖,我在 stackoverflow 看到這樣一個(gè)小技巧

import json

def unique_dicts(data_list: list):
    """unique a list of dict
        dict 是 unhashable 的确徙,不能放入 set 中,所以先轉(zhuǎn)換成json

        unique_dicts([{'a': 1}, {'a': 1}, {'b': 2}])  ->  [{'a': 1}, {'b': 2}]
    """
    data_json_set = set(json.dumps(item) for item in data_list)
    return [json.loads(item) for item in data_json_set]

慢慢更新执桌,想到啥就加啥鄙皇。

本文允許轉(zhuǎn)載,但要求附上源地址 https://www.cnblogs.com/kirito-c/p/9610179.html

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末仰挣,一起剝皮案震驚了整個(gè)濱河市伴逸,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌膘壶,老刑警劉巖错蝴,帶你破解...
    沈念sama閱讀 206,126評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡彻犁,警方通過(guò)查閱死者的電腦和手機(jī)绰姻,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,254評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)官紫,“玉大人躺率,你說(shuō)我怎么就攤上這事⊥蚍” “怎么了悼吱?”我有些...
    開封第一講書人閱讀 152,445評(píng)論 0 341
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)良狈。 經(jīng)常有香客問(wèn)我后添,道長(zhǎng),這世上最難降的妖魔是什么薪丁? 我笑而不...
    開封第一講書人閱讀 55,185評(píng)論 1 278
  • 正文 為了忘掉前任遇西,我火速辦了婚禮,結(jié)果婚禮上严嗜,老公的妹妹穿的比我還像新娘粱檀。我一直安慰自己,他們只是感情好漫玄,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,178評(píng)論 5 371
  • 文/花漫 我一把揭開白布茄蚯。 她就那樣靜靜地躺著,像睡著了一般睦优。 火紅的嫁衣襯著肌膚如雪渗常。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 48,970評(píng)論 1 284
  • 那天汗盘,我揣著相機(jī)與錄音皱碘,去河邊找鬼。 笑死隐孽,一個(gè)胖子當(dāng)著我的面吹牛癌椿,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播菱阵,決...
    沈念sama閱讀 38,276評(píng)論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼踢俄,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了送粱?” 一聲冷哼從身側(cè)響起褪贵,我...
    開封第一講書人閱讀 36,927評(píng)論 0 259
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后脆丁,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體世舰,經(jīng)...
    沈念sama閱讀 43,400評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,883評(píng)論 2 323
  • 正文 我和宋清朗相戀三年槽卫,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了跟压。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 37,997評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡歼培,死狀恐怖震蒋,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情躲庄,我是刑警寧澤查剖,帶...
    沈念sama閱讀 33,646評(píng)論 4 322
  • 正文 年R本政府宣布,位于F島的核電站噪窘,受9級(jí)特大地震影響笋庄,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜倔监,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,213評(píng)論 3 307
  • 文/蒙蒙 一直砂、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧浩习,春花似錦静暂、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,204評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至弯院,卻和暖如春辱士,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背听绳。 一陣腳步聲響...
    開封第一講書人閱讀 31,423評(píng)論 1 260
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留异赫,地道東北人椅挣。 一個(gè)月前我還...
    沈念sama閱讀 45,423評(píng)論 2 352
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像塔拳,于是被迫代替她去往敵國(guó)和親鼠证。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,722評(píng)論 2 345

推薦閱讀更多精彩內(nèi)容