2

五个节约生命的Python小技巧

 2 years ago
source link: https://developer.51cto.com/article/712071.html
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client
214f80339987d311e33728ed771bb2a9ce09d9.jpg

根据笔者经验,下面介绍常用的5个python小技巧:

  1. 字符串操作
  2. lambda 及 map() 函数
  3. if、elif和else单行表达式
  4. 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()方法,''(空格)连接反转列表中的所有单词,并加上一个惊叹号!。

二. 列表推导

c5895738462bed785cb005583b4d4558572e37.jpg

列表推导,一个可以改变你世界观的技巧!这是一个非常强大、直观和可读的方法,可以对列表进行快速操作。

假设,有一个随机的函数,返回一个数字的平方并加上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个快速小技巧,希望对你有用。


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK