最开始学 Python 时,如果我能掌握这些方法,那么代码看起来会更加优美。
选自towardsdatascience,作者:Peter Nistru,机器之心编译,参与:思。
以下即五种实用的小技巧:
字符串运算
列表推导式
lambda 和 map()
单行条件语句
zip()
字符串运算
>>> my_string = "Hi Medium..!">>> print(my_string * 2)Hi Medium..!Hi Medium..!>>> print(my_string + " I love Python" * 2)Hi Medium..! I love Python I love Python复制代码
>>> print(my_string[::-1])!..muideM 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!复制代码
列表推导式
如果你还不会使用列表推导式,那么快学起来吧。正如本文作者所言,「当我第一次学到这种方式时,我的整个世界都变了。」列表推导式真的非常强大,它不仅在速度上比一般的方法快,同时直观性、可读性都非常强。如果你希望迭代列表做一些运算,那么快使用它吧。
我们先定义一个简单的函数,它会算变量的平方并加 5:
>>> 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]复制代码
>>> my_list = [1, 2, 3, 4, 5]>>> print([stupid_func(x) for x in my_list if x % 2 != 0])[6, 14, 30]复制代码
>>> for item in list:>>> if conditional:>>> expression复制代码
>>> print([x ** 2 + 5 for x in my_list if x % 2 != 0])[6, 14, 30]复制代码
Lambda 和 Map
一般而言,Lambda 函数都比较小,它也不需要定义函数名。那么为什么需要匿名函数?简单而言,Lambda 最常执行一些直观的运算,它并不需要标准的函数定义,而且也不需要新的函数名再次调用。
>>> stupid_func = (lambda x : x ** 2 + 5)>>> print([stupid_func(1), stupid_func(3), stupid_func(5)])[6, 14, 30]复制代码
>>> my_list = [2, 1, 0, -1, -2]>>> print(sorted(my_list))[-2, -1, 0, 1, 2]复制代码
>>> print(sorted(my_list, key = lambda x : x ** 2))[0, -1, 1, -2, 2]复制代码
>>> 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]复制代码
如果我们使用条件语句,那么最可能写成下面这个样子:
>>> x = int(input())>>> if x >= 10:>>> print("Horse")>>> elif 1 < x < 10:>>> print("Duck")>>> else:>>> print("Baguette")复制代码
print("Horse" if x >= 10 else "Duck" if 1 < x < 10 else "Baguette")复制代码
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']复制代码