26 Jan 2018
python技巧分享(二)
这是一个系列文章,主要分享python的使用建议和技巧,每次分享3点,希望你能有所收获。
1 如何在if语句中检测多个条件
- 不推荐方式
flag1 = 1
flag2 = 0
flag3 = 0
if flag1 == 1 or flag2 == 1 or flag3 == 1:
print 'ok'
- 推荐方式
flag1 = 1
flag2 = 0
flag3 = 0
if 1 in (flag1, flag2, flag3):
print 'ok'
可以看到,前一种方式重复代码太多,不推荐。后一种写法更简洁、易读。另外,其实也可以通过any函数实现,如下:
if any((flag1, flag2, flag3)):
print 'ok'
2 如何从list中随机采样元素
import random
all_data = [num for num in range(100)]
sample_data = random.sample(all_data, 20)
print sample_data
# [71, 25, 58, 79, 70, 69, 93, 29, 11, 30, 40, 81, 12, 48, 91, 21, 83, 42, 38, 31]
这里是从100个元素中,随机采样20个元素。可以看到,借助random模块的sample函数可以很轻易实现采样功能。python的random模块很有意思,大家感兴趣可以看看,能实现很多有趣的功能。比如可以轻易实现“洗牌”功能:
import random
order_list = range(10)
print order_list
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
random.shuffle(order_list)
print order_list
# [3, 5, 9, 0, 8, 7, 1, 2, 6, 4]
3 如何测试函数运行时间
- 不推荐方式
import time
def func1():
time.sleep(2)
begin_time = time.time()
func1()
passed_time = time.time() - begin_time
print passed_time
# 2.00099992752
- 推荐方式
import time
from timeit import timeit
def func1():
time.sleep(2)
print timeit(stmt="func1()",
setup="from __main__ import func1",
number=1)
# 2.00122451254
借助python的timeit模块,可以很轻易的测试出函数运行时间。在不知道timeit模块之前,我就是用前一种方式测试函数运行时间的,别笑。
LEo
at 22:13