我寫(xiě)的 Python 代碼，同事都說(shuō)好
人生苦短，我用 Python。
程序員的追求就是不寫(xiě)代碼，早日財(cái)務(wù)自由。不對(duì)，一不小心把實(shí)話說(shuō)出來(lái)了，應(yīng)該是將代碼寫(xiě)得簡(jiǎn)潔，優(yōu)雅。
Python 程序員的追求則是 Pythonic，正好在 Python 這門(mén)語(yǔ)言中，「隱藏」了特別多方法，可以使代碼變得簡(jiǎn)潔，優(yōu)雅，與眾不同。
我在這里總結(jié)了一些常用操作，特別是關(guān)于列表和字典，分享給大家。
第一個(gè)字母大寫(xiě)
這個(gè)方法有點(diǎn)意思，無(wú)意中發(fā)現(xiàn)的。
>>> s = "programming is awesome"
>>> print(s.title())
Programming Is Awesome
列表合并
第一種方式：使用 +。
>>> a + b
[1, 2, 3, 4, 5, 6]
第二種方式：使用 extend 關(guān)鍵字。
>>> a.extend(b)
>>> a
[1, 2, 3, 4, 5, 6]
后兩種方式明顯更加優(yōu)雅，推薦使用。需要說(shuō)明的一點(diǎn)是，如果列表很大的話，+ 會(huì)比較慢，使用 extend 更好。
列表元素去重
使用 set() 對(duì)列表元素進(jìn)行去重。
>>> a = [1, 2, 3, 4, 2, 3]
>>> list(set(a))
[1, 2, 3, 4]
列表排序
使用 sort() 或內(nèi)建函數(shù) sorted() 對(duì)列表進(jìn)行排序。它們之間的區(qū)別有兩點(diǎn)：
- sort() 方法是對(duì)原列表進(jìn)行操作，而 sorted() 方法會(huì)返回一個(gè)新列表，不是在原來(lái)的基礎(chǔ)上進(jìn)行操作。
- sort() 是應(yīng)用在列表上的方法，而 sorted() 可以對(duì)所有可迭代的對(duì)象進(jìn)行排序操作。
# sort()
>>> a = [1, 2, 3, 4, 2, 3]
>>> a.sort()
>>> a
[1, 2, 2, 3, 3, 4]
>>>
>>> a = [1, 2, 3, 4, 2, 3]
>>> a.sort(reverse=True)
>>> a
[4, 3, 3, 2, 2, 1]
# sorted()
>>> a = [1, 2, 3, 4, 2, 3]
>>> sorted(a)
[1, 2, 2, 3, 3, 4]
>>> a = [1, 2, 3, 4, 2, 3]
>>> sorted(a, reverse=True)
[4, 3, 3, 2, 2, 1]
遍歷列表的索引和元素對(duì)
使用 enumerate() 函數(shù)可以同時(shí)輸出索引和元素值。
>>> a = ['python', 'go', 'java']
>>> for i, v in enumerate(a):
... print(i, v)
# output
0 python
1 go
2 java
查找列表中出現(xiàn)最頻繁的元素
使用 max() 函數(shù)可以快速查找出一個(gè)列表中出現(xiàn)頻率最高的某個(gè)元素。
>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]
>>> b = max(set(a), key=a.count)
>>> b
4
需要說(shuō)明的一點(diǎn)是，當(dāng)列表中有兩個(gè)元素出現(xiàn)的次數(shù)相同時(shí)，會(huì)返回第一個(gè)出現(xiàn)的元素。
>>> a = [1, 2]
>>> b = max(set(a), key=a.count)
>>> b
1
統(tǒng)計(jì)列表中所有元素的出現(xiàn)次數(shù)
前面的代碼給出了出現(xiàn)最頻繁的值。如果想要知道列表中所有元素的出現(xiàn)次數(shù)，那么可以使用 collections 模塊。
collections 是 Python 中的一個(gè)寶藏模塊，它提供了很多特性。Counter 方法正好可以完美解決這個(gè)需求。
>>> from collections import Counter
>>>
>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]
>>> Counter(a)
Counter({4: 4, 2: 2, 3: 2, 1: 1, 5: 1})
將兩個(gè)列表合并為字典
使用 zip() 函數(shù)，可以將兩個(gè)列表合并成字典。
>>> a = ['one', 'tow', 'three']
>>> b = [1, 2, 3]
>>> dict(zip(a, b))
{'one': 1, 'tow': 2, 'three': 3}
求兩個(gè)列表的交集，并集和差集。
# list_operate.py
def main():
list_a = [1, 2, 3, 4, 5]
list_b = [4, 5, 6, 7, 8]
# 求交集的兩種方式
res_a = [i for i in list_a if i in list_b]
res_b = list(set(list_a).intersection(set(list_b)))
print(f"res_a is: {res_a}")
print(f"res_b is: {res_b}")
# 求并集
res_c = list(set(list_a).union(set(list_b)))
print(f"res_c is: {res_c}")
# 求差集的兩種方式，在B中但不在A中
res_d = [i for i in list_b if i not in list_a]
res_e = list(set(list_b).difference(set(list_a)))
print(f"res_d is: {res_d}")
print(f"res_e is: {res_e}")
if __name__ == '__main__':
main()
字典創(chuàng)建
# 1、創(chuàng)建空字典
a = {}
b = dict()
# 2、有初始值，從輸入的便利程度來(lái)說(shuō)，我更喜歡第二種
a = {'a': 1, 'b': 2, 'c': 3}
b = dict(a=1, b=2, c=3)
# 3、key 來(lái)自一個(gè)列表，而 value 相同, 使用 fromkeys，那是相當(dāng)?shù)膬?yōu)雅
keys = ['a', 'b', 'c']
value = 100
d = dict.fromkeys(keys, value)
# 4、key 來(lái)自一個(gè)列表，而 value 也是一個(gè)列表，使用 zip
keys = ['a', 'b', 'c']
values = [1, 2, 3]
d = dict(zip(keys, values))
字典合并
m = {'a': 1}
n = {'b': 2, 'c': 3}
# 合并，兩種方式
# 1、使用 update
m.update(n)
# 2、使用 **
{**m, **n}判斷 key 是否存在
在 Python2 中判斷某個(gè) key 是否存在，可以使用 has_key，但這個(gè)方法在 Python3 中已經(jīng)被移除了。
另一種方法是使用 in 關(guān)鍵字，不僅兼容 Python2 和 Python3，速度還更快，強(qiáng)烈推薦。
d = {'a': 1, 'b': 2}
if 'a' in d:
print('hello') 獲取字典中的值
d = {'a': 1, 'b': 2}
# 1、直接用 key 取值，但這種方式不好，如果 key 不存在會(huì)報(bào)錯(cuò)，推薦使用 get
a = d['a']
# 2、使用 get，如果 key 不存在還可以賦默認(rèn)值
a = d.get('a')
c = d.get('c', 3)字典遍歷
d = {'a': 1, 'b': 2, 'c': 3}
# 遍歷 key
for key in d.keys():
pass
# 遍歷 value
for value in d.values():
pass
# 遍歷 key 和 value
for key, value in d.items():
pass字典推導(dǎo)式
列表推導(dǎo)式和字典推導(dǎo)式是我相當(dāng)喜歡的功能，簡(jiǎn)潔高效。map 和 filter 我都已經(jīng)快不會(huì)用了。
l = [1, 2, 3]
{n: n * n for n in l}
{1: 1, 2: 4, 3: 9}
字典按 key 或 value 排序
d = {'a': 1, 'b': 2, 'e': 9, 'c': 5, 'd': 7}
# 按 key 排序
sorted(d.items(), key=lambda t: t[0])
# 按 key 倒序
sorted(d.items(), key=lambda t: t[0], reverse=True)
# 按 value 排序
sorted(d.items(), key=lambda t: t[1])還有一個(gè)需求是我在開(kāi)發(fā)過(guò)程經(jīng)常碰到的，就是有一個(gè)列表，列表的元素是字典，然后按字典的 value 對(duì)列表進(jìn)行排序。
l = [{'name': 'a', 'count': 4}, {'name': 'b', 'count': 1}, {'name': 'd', 'count': 2}, {'name': 'c', 'count': 6}]
sorted(l, key=lambda e: e.__getitem__('count'))
# 倒序
sorted(l, key=lambda e: e.__getitem__('count'), reverse=True)
