python技巧分享（九）

這是一個系列文章，主要分享python的使用建議和技巧，每次分享3點，希望你能有所收穫。

1 正確打開文件

• 不推薦方式

f = open("tmp")
for line in f.readlines():
    print line

f.close()

'''
line 1

line 2

line 3
'''

• 推薦方式

with open("tmp") as f:
    for line in f.readlines():
        print line

'''
line 1

line 2

line 3
'''

前一種方式需要顯式的調用close()方法關閉打開的文件，如果因爲異常或者其他原因沒有關閉打開的文件，會導致資源泄露，這是一種很不好的編程習慣。後一種方式使用with語句方式打開文件，文件在with語句塊範圍外會自動關閉，不需要調用close()方法關閉文件。

2 創建有序字典

• 推薦方式

import collections

ordered_dict = collections.OrderedDict()
ordered_dict['0'] = 0
ordered_dict['1'] = 1
ordered_dict['2'] = 2
ordered_dict['3'] = 3
ordered_dict['4'] = 4

print "ordered dict:"
for key, value in ordered_dict.items():
    print "ordered_dict[{key}] = {value}".format(key=key, value=value)

regular_dict = {}
regular_dict['0'] = 0
regular_dict['1'] = 1
regular_dict['2'] = 2
regular_dict['3'] = 3
regular_dict['4'] = 4

print "\nregular dict:"
for key, value in regular_dict.items():
    print "regular_dict[{key}] = {value}".format(key=key, value=value)

'''
ordered dict:
ordered_dict[0] = 0
ordered_dict[1] = 1
ordered_dict[2] = 2
ordered_dict[3] = 3
ordered_dict[4] = 4

regular dict:
regular_dict[1] = 1
regular_dict[0] = 0
regular_dict[3] = 3
regular_dict[2] = 2
regular_dict[4] = 4
'''

由於字典存儲時是無序的，如果需要將存儲順序記錄下來，可以通過使用collections.OrderedDict()方式實現，從上面示例可以看到，對於一般字典，遍歷打印該字典時，不是按存儲順序輸出。如果使用有序字典，打印時就會按存儲的順序輸出。

3 模擬switch語句

• 不推薦方式

def calc(operator, x, y):
        if operator == '+':
            return x + y
        elif operator == '-':
            return x - y
        elif operator == '*':
            return x * y
        elif operator == '/':
            return x / y
        else:
            return 'invalid operator'


print calc('+', 2, 1)
# 3
print calc('-', 2, 1)
# 1
print calc('*', 2, 1)
# 2
print calc('/', 2, 1)
# 2
print calc('&', 2, 1)
# invalid operator

• 推薦方式

def calc(operator, x, y):
    operator_dict = {
        '+': x + y,
        '-': x - y,
        '*': x * y,
        '/': x / y,
    }

    result = operator_dict.get(operator, None)
    if result is None:
        return 'invalid operator'
    else:
        return result


print calc('+', 2, 1)
# 3
print calc('-', 2, 1)
# 1
print calc('*', 2, 1)
# 2
print calc('/', 2, 1)
# 2
print calc('&', 2, 1)
# invalid operator

前一種方式中，通過if else方式實現類似switch的功能，但是代碼比較多，尤其重複代碼。後一種方式藉助字典的特性，輕易實現switch功能，代碼量更少且易讀。