五個節(jié)約生命的Python小技巧
根據筆者經驗,下面介紹常用的5個python小技巧:
- 字符串操作
- 列表推導
- lambda 及 map() 函數(shù)
- if、elif和else單行表達式
- zip()函數(shù)
一. 字符串操作
Python善于用數(shù)學運算符(如+和*)對字符串進行操作:
- + 拼接字符串
- * 重復字符串
my_string = "Hi Python..!"
print(my_string * 2)
#Hi Python..!Hi Python..!
print(my_string + " I love Python" * 2)
#Hi Python..! I love Python I love Python
也可以用切片操作[::-1]輕松反轉一個字符串,并且不限于字符串(如列表翻轉)!
my_string = "Hi Python..!"
print(my_string[::-1])
# !..nohtyP iH
my_list = [1,2,3,4,5]
print(my_list[::-1])
# [5, 4, 3, 2, 1]
下面是對一個單詞列表進行了反轉拼接成字符串:
word_list = ["awesome", "is", "this"]
print(' '.join(word_list[::-1]) + '!')
#this is awesome!
用.join()方法,''(空格)連接反轉列表中的所有單詞,并加上一個驚嘆號!。
二. 列表推導
列表推導,一個可以改變你世界觀的技巧!這是一個非常強大、直觀和可讀的方法,可以對列表進行快速操作。
假設,有一個隨機的函數(shù),返回一個數(shù)字的平方并加上5:
def stupid_func(x):
return x**2 + 5
現(xiàn)在,想把函數(shù)stupid_func()應用于列表中的所有奇數(shù),如果不用列表推導,笨辦法如下:
def stupid_func(x):
return x**2 + 5
my_list = [1, 2, 3, 4, 5]
new_list = []
for x in my_list:
if x % 2 != 0:
new_list.append(stupid_func(x))
print(new_list)
#[6, 14, 30]
如果用列表推導,代碼瞬間變的優(yōu)雅:
def stupid_func(x):
return x**2 + 5
my_list = [1, 2, 3, 4, 5]
print([stupid_func(x) for x in my_list if x % 2 != 0])
#[6, 14, 30]
列表推導的語法:[ expression for item in list ],如果覺得不夠花哨,還可以加上一個判斷條件,比如上面的"奇數(shù)"條件: [expression for item in list if conditional]。本質上如下代碼的功能:
for item in list:
if conditional:
expression
Very Cool!。不過還可以更進一步,直接省去stupid_func()函數(shù):
my_list = [1, 2, 3, 4, 5]
print([x ** 2 + 5 for x in my_list if x % 2 != 0])
#[6, 14, 30]
三. Lambda & Map函數(shù)
Lambda
Lambda看上去有點點奇怪,但奇怪的東西一般功能都很強大,一旦你掌握就很直觀,省去大量廢話代碼。
基本上,Lambda函數(shù)是一個小型的匿名函數(shù)。為什么是匿名的?
因為Lambda最常被用來執(zhí)行的簡單操作,但不需要像def my_function()那樣正兒八經,所以Lambda又名吊兒郎當函數(shù)(瞎編的,忽略忽略)。
改進上面的例子:def stupid_func(x)可以用一行Lambda函數(shù)來代替:
stupid_func = (lambda x : x ** 2 + 5)
print([stupid_func(1), stupid_func(3), stupid_func(5)])
#[6, 14, 30]
那么為什么要使用這種奇怪的語法呢?當想進行一些簡單操作而不需要定義實際函數(shù)時,這就變得很有用。
以一個數(shù)字列表為例。假設對列表進行排序?一種方法是使用 sorted() 方法:
my_list = [2, 1, 0, -1, -2]
print(sorted(my_list))
#[-2, -1, 0, 1, 2]
sorted()函數(shù)可以完成排序,但假設想按每個數(shù)的平方進行排序呢?此時可用lambda函數(shù)來定義排序鍵key,這也是sorted()方法用來決定如何排序的:
my_list = [2, 1, 0, -1, -2]
print(sorted(my_list, key = lambda x : x ** 2))
#[0, -1, 1, -2, 2]
Map函數(shù)
map是python內置函數(shù),會根據提供的函數(shù)對指定的序列做映射。假設有一個列表,想把列表中的每個元素與另一個列表中的相應元素相乘,如何做到這一點?使用lambda函數(shù)和map!
print(list(map(lambda x, y : x * y, [1, 2, 3], [4, 5, 6])))
#[4, 10, 18]
與下面這種常規(guī)廢話代碼,簡單而優(yōu)雅:
x, y = [1, 2, 3], [4, 5, 6]
z = []
for i in range(len(x)):
z.append(x[i] * y[i])
print(z)
#[4, 10, 18]
四. if-else 單行表達
在你的代碼的某個地方,可能會有這樣廢話的條件語句:
x = int(input())
if x >= 10:
print("Horse")
elif 1 < x < 10:
print("Duck")
else:
print("Baguette")
當運行程序時,提示從input()函數(shù)中輸入一個信息,比如輸入5,得到Duck。但其實也可以一行代碼完成整個事情:
print("Horse" if x >= 10 else "Duck" if 1 < x < 10 else "Baguette")
一行代碼簡單直接! 翻閱你的舊代碼,會發(fā)現(xiàn)很多判斷都可以規(guī)約為一個if-else單行表達式。
五. zip()函數(shù)
還記得map()函數(shù)部分兩個列表元素按位相乘嗎?
zip()使之更加簡單。假設有兩個列表,一個包含名,一個包含姓,如何有序地合并它們呢?使用zip()!
first_names = ["Peter", "Christian", "Klaus"]
last_names = ["Jensen", "Smith", "Nistrup"]
print([' '.join(x) for x in zip(first_names, last_names)])
#['Peter Jensen', 'Christian Smith', 'Klaus Nistrup']
小節(jié)
上面列出的5個快速小技巧,希望對你有用。