根据笔者经验,下面介绍常用的5个python小技巧:
- 字符串操作
- 列表推导
- lambda 及 map() 函数
- if、elif和else单行表达式
- zip()函数
一. 字符串操作
Python善于用数学运算符(如+和*)对字符串进行操作:
- + 拼接字符串
- * 重复字符串
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()方法,''(空格)连接反转列表中的所有单词,并加上一个惊叹号!。
二. 列表推导
列表推导,一个可以改变你世界观的技巧!这是一个非常强大、直观和可读的方法,可以对列表进行快速操作。
假设,有一个随机的函数,返回一个数字的平方并加上5:
def stupid_func(x):
return x**2 + 5
现在,想把函数stupid_func()应用于列表中的所有奇数,如果不用列表推导,笨办法如下:
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]
如果用列表推导,代码瞬间变的优雅:
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 ],如果觉得不够花哨,还可以加上一个判断条件,比如上面的"奇数"条件: [expression for item in list if conditional]。本质上如下代码的功能:
for item in list:
if conditional:
expression
Very Cool!。不过还可以更进一步,直接省去stupid_func()函数:
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函数
Lambda
Lambda看上去有点点奇怪,但奇怪的东西一般功能都很强大,一旦你掌握就很直观,省去大量废话代码。
基本上,Lambda函数是一个小型的匿名函数。为什么是匿名的?
因为Lambda最常被用来执行的简单操作,但不需要像def my_function()那样正儿八经,所以Lambda又名吊儿郎当函数(瞎编的,忽略忽略)。
改进上面的例子:def stupid_func(x)可以用一行Lambda函数来代替:
stupid_func = (lambda x : x ** 2 + 5)
print([stupid_func(1), stupid_func(3), stupid_func(5)])
#[6, 14, 30]
那么为什么要使用这种奇怪的语法呢?当想进行一些简单操作而不需要定义实际函数时,这就变得很有用。
以一个数字列表为例。假设对列表进行排序?一种方法是使用 sorted() 方法:
my_list = [2, 1, 0, -1, -2]
print(sorted(my_list))
#[-2, -1, 0, 1, 2]
sorted()函数可以完成排序,但假设想按每个数的平方进行排序呢?此时可用lambda函数来定义排序键key,这也是sorted()方法用来决定如何排序的:
my_list = [2, 1, 0, -1, -2]
print(sorted(my_list, key = lambda x : x ** 2))
#[0, -1, 1, -2, 2]
Map函数
map是python内置函数,会根据提供的函数对指定的序列做映射。假设有一个列表,想把列表中的每个元素与另一个列表中的相应元素相乘,如何做到这一点?使用lambda函数和map!
print(list(map(lambda x, y : x * y, [1, 2, 3], [4, 5, 6])))
#[4, 10, 18]
与下面这种常规废话代码,简单而优雅:
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()函数中输入一个信息,比如输入5,得到Duck。但其实也可以一行代码完成整个事情:
print("Horse" if x >= 10 else "Duck" if 1 < x < 10 else "Baguette")
一行代码简单直接! 翻阅你的旧代码,会发现很多判断都可以规约为一个if-else单行表达式。
五. zip()函数
还记得map()函数部分两个列表元素按位相乘吗?
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']
小节
上面列出的5个快速小技巧,希望对你有用。