年會抽獎如何用 Python讓自己變成天選之子
請大家猜一猜下面這段代碼的運行效果:
- import random
- import time
- people = ['kingname', '王小一', '李小二', '張小三', '劉小四', '盧小五', '馬小六', '周小七', '丁小八', '朱小九']
- for i in range(1, 11):
- lucky_guy = random.choice(people)
- print(f'第{i}次抽獎,中獎用戶:{lucky_guy}')
- time.sleep(1)
你是不是以為這段代碼運行以后,結果如下圖所示?
但實際上,我可以讓輸出結果根據我的意愿隨意變動,例如像下面這個 gif ,所有輸出結果都是我:
你可以先不要往下看,放下手機,自己寫一下代碼,試一試 如何才能實現 gif 中的效果。
下面來為大家解密。
要實現這個效果,只需要兩個知識點:
- Python自帶模塊是可以被覆蓋的
- Python 的 import 在同一個運行時只會導入一次
首先來看第一個知識點。Python 的自帶模塊是可以被覆蓋的,所以我們先來定義一個函數:
- def choice(option):
- return 'kingname'
接下來,使用這個函數覆蓋random.choice:
- import random
- random.choice = choice
現在,無論給random.choice傳入什么參數,它始終都會返回kingname,運行效果如下圖所示:
這個時候,你可能會說,那別人寫代碼的時候,又重新import random怎么辦呢?random.choice不是又被改回去了嗎?
實際上并不會,因為Python 的包導入機制決定了,在每個運行時內部,每個包只有第一次導入的時候有效,所以只要還在當前運行時,那么后續的所有import random都是無效的。
所以,即使重新導入了 random 模塊,random.choice依然是你修改以后的代碼。所以當你再次執行的時候,會發現返回的還是你想要的數據,如下圖所示:
可能有人會說這樣容易被識破啊,別人只要先隨便寫一些測試數據,運行一次random.choice([123, 456]),發現返回的竟然是kingname,這不就露餡了嗎?
實際上完全不用擔心,我們可以這樣操作:
如果備選列表里面不包含kingname,那么就使用原生的 random.choice
如果備選列表里面包含kingname,那么就以60%的概率返回kingname
要實現這樣的功能,我們可以這樣寫代碼:
首先重啟當前 Jupyter 內核,讓 random 恢復成默認的,然后編碼:
- import random
- origin_choice = random.choice
- def choice(option):
- if 'kingname' not in option or random.randint(1, 10) > 6:
- return origin_choice(option)
- return 'kingname'
- random.choice = choice
這樣替換以后,當有kingname在備選列表中時,kingname被有60%的概率被選中,如下圖所示:
當kingname不在備選列表中時,一切正常,如下圖所示:
本文轉載自微信公眾號「未聞Code」,可以通過以下二維碼關注。轉載本文請聯系未聞Code公眾號。