码迷,mamicode.com
首页 > 编程语言 > 详细

Python代码优化及技巧笔记(二)

时间:2016-03-17 14:48:53      阅读:305      评论:0      收藏:0      [点我收藏+]

标签:

概述
这里是记录一些本人在开发过程中遇到的一些细节及代码优化问题,希望与君共勉。


版权说明

著作权归作者所有。
商业转载请联系作者获得授权,非商业转载请注明出处。
作者:Coding-Naga
发表日期: 2016年3月17日
链接:http://blog.csdn.net/lemon_tree12138/article/details/50854673
来源:CSDN
更多内容:分类 >> Thinking In Python


目录


奇技淫巧


1. Python代码获取命令行输出

这里可以通过subprocess模块中的Popen, PIPE来实现。
比如有这如下代码,就可以尝试捕获控制台输出的文本信息:

from subprocess import Popen, PIPE

label = "Hello, Shell."
print(label)

f = Popen(("python", "catch_output.py"), stdout=PIPE).stdout
print("Catch Output: {0}".format(f.readline()))

上面的代码中我们可以打印出自身的控制台输出,还有通过捕获输出获得的输出信息。如下:

Hello, Shell.
Catch Output: Hello, Shell.

2. 不要将表达式作为函数的默认参数

在Python有一个比较基础也是比较重要的特性,那就是在python的函数中,我们可以给参数指定一个默认的值。但是,这是有一个小小的陷阱。对于新手来说可能会经常引起困扰,比如我们像下面这样编写python代码:

def test_args(array=[]):
    array.append("Bob")
    return array

以上的代码中正常情况下,是没有什么问题的,因为的的确确可以打印出“[Bob]”。可是,问题是出现在我们重复的调用上。怎么说?就以上面的test_args()方法,我们进行三次重复调用。打印的结果却是:

[‘Bob‘]
[‘Bob‘, ‘Bob‘]
[‘Bob‘, ‘Bob‘, ‘Bob‘]

WTF!
这是怎么会回事呢?因为,可选参数默认值的设置在Python中只会被执行一次
怎么来理解?很简单,也就是我们的参数默认值只有在对此函数进行定义,并且在函数调用的时候不传递此参数的值的时候,才被认为需要对其进行默认值的赋值。
想要修改此函数也很简单,代码如下:

def test_args(array=None):
    if array is None:
        array = []
    array.append("Bob")
    return array

3. 指定异常代码块(exception block)的参数

在python2.x中,我们知道可以使用逗号来进行异常参数的指定。如下:

def fun():
    try:
        print("Hello, Exception")
    except Exception, e:
        print(e)
    pass

可是在python3.x中,这种写法却存在着语法上的错误。不过对于python2.x与python3.x来说,都可以使用as来进行指定。如下:

def fun():
    try:
        print("Hello, Exception")
    except Exception as e:
        print(e)
    pass

所以,对于可能存在版本差异的项目来说,最好还是使用as来指定参数更为妥当。


4. 在遍历列表时更改列表

正常情况下我们很难在遍历列表的过程中能列表进行修改,这一条不仅在Python中适用,在Java中也存在着同样的规则。比如,如下的写法就会抛出一些异常。

def test_list_modify():
    a = [0, 1, 2, 3, 4, 5, 6, 7, 8]
    odd = lambda x: bool(x % 2)
    for i in xrange(len(a)):
        if odd(a[i]):
            a.remove(i)
    print(a)

上面的代码一定会抛出异常,异常如下所示:

Traceback (most recent call last):
  File "E:/workspace/src/Python/Demo/SimpleDemo-python/test/test_demo.py", line 106, in <module>
    test_list_modify()
  File "E:/workspace/src/Python/Demo/SimpleDemo-python/test/test_demo.py", line 76, in test_list_modify
    if odd(a[i]):
IndexError: list index out of range

原因也很容易找到,上面的代码中,我们尝试去修改列表的长度,这样会导致循环的过程中,访问数组的下标会超出修改后的列表长度。从而抛出以上异常信息。
不过,我们可以利用Python语言自身优雅的编程范式。修改后的代码如下:

def test_list_modify():
    a = [0, 1, 2, 3, 4, 5, 6, 7, 8]
    odd = lambda x: bool(x % 2)
    a[:] = [n for n in a if not odd(n)]
    print(a)

5. 更优雅地打印出JSON

在Python中,json模块为我们提供了非常好的json对象的打印接口——json.dumps()
当然json.dumps()中传入的是json这个对象,而不是原始的json字符串。想要实现将原始json字符串优雅地打印出来,我们可以对其进行二次封装。

  1. 通过json.loads()将json字符串转化成json对象;
  2. 通过json.dumps()将json字符串优雅地打印出来.
import json

json_data = "{\"status\": \"OK\", \"count\": 2, \"results\": [{\"age\": 27, \"name\": \"Oz\", \"lactose_intolerant\": true},"             " {\"age\": 29, \"name\": \"Joe\", \"lactose_intolerant\": false}]}"

def parser():
    json_obj = json.loads(json_data)
    show(json_obj)
    pass

def show(json_obj):
    print(json.dumps(json_obj, indent=4))
    pass

if __name__ == ‘__main__‘:
    parser()
    pass

打印的结果如下:
技术分享


6. __init__.py的功能

有时我们需要向一个自定义模块中导入很多其他模块的内容,这样会让代码显得有一些臃肿。不过,好在我们可以通过__init__.py这个文件来解决这个问题。做法是将导入模块的代码放在__init__.py这个文件中,再在目标文件中导入__init__.py这一个模块中的全部内容即可。
在__init__.py中有如下代码:

import os
import time
import datetime as date

测试文件中的调用代码如下:

from __init__ import *

print(time.ctime())
print(date.date.day)
print(os.system("python test_init.py"))

当然,你也可以选择一个一个地导入,如下:

from __init__ import sys
from __init__ import getopt

from __init__ import pack
from __init__ import pehelp
from __init__ import PEParser
from __init__ import Signature

Ref

http://codingpy.com/article/top-10-mistakes-that-python-programmers-make/
http://www.vaikan.com/improving-your-python-productivity/


Python代码优化及技巧笔记(二)

标签:

原文地址:http://blog.csdn.net/lemon_tree12138/article/details/50854673

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!