Python中非常實(shí)用的一些功能和函數(shù)分享

更新時(shí)間：2015年02月14日 09:53:19 投稿：junjie

這篇文章主要介紹了Python中非常實(shí)用的一些功能和函數(shù)分享,本文講解了帶任意數(shù)量參數(shù)的函數(shù)、使用Glob()查找文件、調(diào)試、生成唯一ID等內(nèi)容,需要的朋友可以參考下

在使用Python多年以后，我偶然發(fā)現(xiàn)了一些我們過(guò)去不知道的功能和特性。一些可以說(shuō)是非常有用，但卻沒(méi)有充分利用?？紤]到這一點(diǎn)，我編輯了一些你應(yīng)該了解的Python功能特色。

帶任意數(shù)量參數(shù)的函數(shù)
你可能已經(jīng)知道了Python允許你定義可選參數(shù)。但還有一個(gè)方法，可以定義函數(shù)任意數(shù)量的參數(shù)。
首先，看下面是一個(gè)只定義可選參數(shù)的例子

復(fù)制代碼代碼如下:

def function(arg1="",arg2=""):
print "arg1: {0}".format(arg1)
print "arg2: {0}".format(arg2)

function("Hello", "World")
# prints args1: Hello
# prints args2: World

function()
# prints args1:
# prints args2:

現(xiàn)在，讓我們看看怎么定義一個(gè)可以接受任意參數(shù)的函數(shù)。我們利用元組來(lái)實(shí)現(xiàn)

復(fù)制代碼代碼如下:

def foo(*args): # just use "*" to collect all remaining arguments into a tuple
    numargs = len(args)
    print "Number of arguments: {0}".format(numargs)
    for i, x in enumerate(args):
        print "Argument {0} is: {1}".format(i,x)

foo()
# Number of arguments: 0

foo("hello")
# Number of arguments: 1
# Argument 0 is: hello

foo("hello","World","Again")
# Number of arguments: 3
# Argument 0 is: hello
# Argument 1 is: World
# Argument 2 is: Again

使用Glob()查找文件
大多Python函數(shù)有著長(zhǎng)且具有描述性的名字。但是命名為glob()的函數(shù)你可能不知道它是干什么的除非你從別處已經(jīng)熟悉它了。
它像是一個(gè)更強(qiáng)大版本的listdir()函數(shù)。它可以讓你通過(guò)使用模式匹配來(lái)搜索文件。

復(fù)制代碼代碼如下:

import glob

# get all py files
files = glob.glob('*.py')
print files

# Output
# ['arg.py', 'g.py', 'shut.py', 'test.py']

你可以像下面這樣查找多個(gè)文件類型：

復(fù)制代碼代碼如下:

import itertools as it, glob

def multiple_file_types(*patterns):
return it.chain.from_iterable(glob.glob(pattern) for pattern in patterns)

for filename in multiple_file_types("*.txt", "*.py"): # add as many filetype arguements
print filename

# output
#=========#
# test.txt
# arg.py
# g.py
# shut.py
# test.py

如果你想得到每個(gè)文件的絕對(duì)路徑，你可以在返回值上調(diào)用realpath()函數(shù)：

復(fù)制代碼代碼如下:

import itertools as it, glob, os

def multiple_file_types(*patterns):
return it.chain.from_iterable(glob.glob(pattern) for pattern in patterns)

for filename in multiple_file_types("*.txt", "*.py"): # add as many filetype arguements
realpath = os.path.realpath(filename)
print realpath

# output
#=========#
# C:\xxx\pyfunc\test.txt
# C:\xxx\pyfunc\arg.py
# C:\xxx\pyfunc\g.py
# C:\xxx\pyfunc\shut.py
# C:\xxx\pyfunc\test.py

調(diào)試

下面的例子使用inspect模塊。該模塊用于調(diào)試目的時(shí)是非常有用的，它的功能遠(yuǎn)比這里描述的要多。

這篇文章不會(huì)覆蓋這個(gè)模塊的每個(gè)細(xì)節(jié)，但會(huì)展示給你一些用例。

復(fù)制代碼代碼如下:

import logging, inspect

logging.basicConfig(level=logging.INFO,
    format='%(asctime)s %(levelname)-8s %(filename)s:%(lineno)-4d: %(message)s',
    datefmt='%m-%d %H:%M',
    )
logging.debug('A debug message')
logging.info('Some information')
logging.warning('A shot across the bow')

def test():
    frame,filename,line_number,function_name,lines,index=\
        inspect.getouterframes(inspect.currentframe())[1]
    print(frame,filename,line_number,function_name,lines,index)

test()

# Should print the following (with current date/time of course)
#10-19 19:57 INFO test.py:9 : Some information
#10-19 19:57 WARNING test.py:10 : A shot across the bow
#(, 'C:/xxx/pyfunc/magic.py', 16, '', ['test()\n'], 0)

生成唯一ID

在有些情況下你需要生成一個(gè)唯一的字符串。我看到很多人使用md5()函數(shù)來(lái)達(dá)到此目的，但它確實(shí)不是以此為目的。其實(shí)有一個(gè)名為uuid()的Python函數(shù)是用于這個(gè)目的的。

復(fù)制代碼代碼如下:

import uuid
result = uuid.uuid1()
print result

# output => various attempts
# 9e177ec0-65b6-11e3-b2d0-e4d53dfcf61b
# be57b880-65b6-11e3-a04d-e4d53dfcf61b
# c3b2b90f-65b6-11e3-8c86-e4d53dfcf61b
你可能會(huì)注意到，即使字符串是唯一的，但它們后邊的幾個(gè)字符看起來(lái)很相似。這是因?yàn)樯傻淖址c電腦的MAC地址是相聯(lián)系的。

為了減少重復(fù)的情況，你可以使用這兩個(gè)函數(shù)。

import hmac,hashlib
key='1'
data='a'
print hmac.new(key, data, hashlib.sha256).hexdigest()

m = hashlib.sha1()
m.update("The quick brown fox jumps over the lazy dog")
print m.hexdigest()

# c6e693d0b35805080632bc2469e1154a8d1072a86557778c27a01329630f8917
# 2fd4e1c67a2d28fced849ee1bb76e7391b93eb12

序列化

你曾經(jīng)需要將一個(gè)復(fù)雜的變量存儲(chǔ)在數(shù)據(jù)庫(kù)或文本文件中吧？你不需要想一個(gè)奇特的方法將數(shù)組或?qū)ο蟾褶D(zhuǎn)化為式化字符串，因?yàn)镻ython已經(jīng)提供了此功能。

復(fù)制代碼代碼如下:

import pickle

variable = ['hello', 42, [1,'two'],'apple']

# serialize content
file = open('serial.txt','w')
serialized_obj = pickle.dumps(variable)
file.write(serialized_obj)
file.close()

# unserialize to produce original content
target = open('serial.txt','r')
myObj = pickle.load(target)

print serialized_obj
print myObj

#output
# (lp0
# S'hello'
# p1
# aI42
# a(lp2
# I1
# aS'two'
# p3
# aaS'apple'
# p4
# a.
# ['hello', 42, [1, 'two'], 'apple']

這是一個(gè)原生的Python序列化方法。然而近幾年來(lái)JSON變得流行起來(lái)，Python添加了對(duì)它的支持?，F(xiàn)在你可以使用JSON來(lái)編解碼。

復(fù)制代碼代碼如下:

import json

variable = ['hello', 42, [1,'two'],'apple']
print "Original {0} - {1}".format(variable,type(variable))

# encoding
encode = json.dumps(variable)
print "Encoded {0} - {1}".format(encode,type(encode))

#deccoding
decoded = json.loads(encode)
print "Decoded {0} - {1}".format(decoded,type(decoded))

# output

# Original ['hello', 42, [1, 'two'], 'apple'] - <type 'list'="">
# Encoded ["hello", 42, [1, "two"], "apple"] - <type 'str'="">
# Decoded [u'hello', 42, [1, u'two'], u'apple'] - <type 'list'="">

這樣更緊湊，而且最重要的是這樣與JavaScript和許多其他語(yǔ)言兼容。然而對(duì)于復(fù)雜的對(duì)象，其中的一些信息可能丟失。

壓縮字符
當(dāng)談起壓縮時(shí)我們通常想到文件，比如ZIP結(jié)構(gòu)。在Python中可以壓縮長(zhǎng)字符，不涉及任何檔案文件。

復(fù)制代碼代碼如下:

import zlib

string = """   Lorem ipsum dolor sit amet, consectetu
                adipiscing elit. Nunc ut elit id mi ultricies
                adipiscing. Nulla facilisi. Praesent pulvinar,
                sapien vel feugiat vestibulum, nulla dui pretium orci,
                non ultricies elit lacus quis ante. Lorem ipsum dolor
                sit amet, consectetur adipiscing elit. Aliquam
                pretium ullamcorper urna quis iaculis. Etiam ac massa
                sed turpis tempor luctus. Curabitur sed nibh eu elit
                mollis congue. Praesent ipsum diam, consectetur vitae
                ornare a, aliquam a nunc. In id magna pellentesque
                tellus posuere adipiscing. Sed non mi metus, at lacinia
                augue. Sed magna nisi, ornare in mollis in, mollis
                sed nunc. Etiam at justo in leo congue mollis.
                Nullam in neque eget metus hendrerit scelerisque
                eu non enim. Ut malesuada lacus eu nulla bibendum
                id euismod urna sodales. """

print "Original Size: {0}".format(len(string))

compressed = zlib.compress(string)
print "Compressed Size: {0}".format(len(compressed))

decompressed = zlib.decompress(compressed)
print "Decompressed Size: {0}".format(len(decompressed))

# output

# Original Size: 1022
# Compressed Size: 423
# Decompressed Size: 1022

注冊(cè)Shutdown函數(shù)

有可模塊叫atexit，它可以讓你在腳本運(yùn)行完后立馬執(zhí)行一些代碼。
假如你想在腳本執(zhí)行結(jié)束時(shí)測(cè)量一些基準(zhǔn)數(shù)據(jù)，比如運(yùn)行了多長(zhǎng)時(shí)間：

復(fù)制代碼代碼如下:

import atexit
import time
import math

def microtime(get_as_float = False) :
    if get_as_float:
        return time.time()
    else:
        return '%f %d' % math.modf(time.time())
start_time = microtime(False)
atexit.register(start_time)

def shutdown():
global start_time
print "Execution took: {0} seconds".format(start_time)

atexit.register(shutdown)

# Execution took: 0.297000 1387135607 seconds
# Error in atexit._run_exitfuncs:
# Traceback (most recent call last):
#   File "C:\Python27\lib\atexit.py", line 24, in _run_exitfuncs
#     func(*targs, **kargs)
# TypeError: 'str' object is not callable
# Error in sys.exitfunc:
# Traceback (most recent call last):
#   File "C:\Python27\lib\atexit.py", line 24, in _run_exitfuncs
#     func(*targs, **kargs)
# TypeError: 'str' object is not callable

打眼看來(lái)很簡(jiǎn)單。只需要將代碼添加到腳本的最底層，它將在腳本結(jié)束前運(yùn)行。但如果腳本中有一個(gè)致命錯(cuò)誤或者腳本被用戶終止，它可能就不運(yùn)行了。
當(dāng)你使用atexit.register()時(shí)，你的代碼都將執(zhí)行，不論腳本因?yàn)槭裁丛蛲Ｖ惯\(yùn)行。

結(jié)論

你是否意識(shí)到那些不是廣為人知Python特性很有用？請(qǐng)?jiān)谠u(píng)論處與我們分享。謝謝你的閱讀！

您可能感興趣的文章: