首頁  >  文章  >  後端開發  >  我寫的 Python 程式碼,同事都說好

我寫的 Python 程式碼,同事都說好

王林
王林轉載
2023-04-12 13:43:121169瀏覽

人生苦短,我用 Python。

程式設計師的追求就是不寫程式碼,早日財務自由。不對,一不小心把實話說出來了,應該是將程式碼寫得簡潔,優雅。

Python 程式設計師的追求則是 Pythonic,正好在 Python 這門語言中,「隱藏」了特別多方法,可以讓程式碼變得簡潔,優雅,與眾不同。

我在這裡總結了一些常用操作,特別是關於清單和字典,分享給大家。

第一個字母大寫

這個方法有點意思,無意中發現的。

>>> s = "programming is awesome"
>>> print(s.title())
Programming Is Awesome

清單合併

第一種方式:使用 。

>>> a + b
[1, 2, 3, 4, 5, 6]

第二種方式:使用 extend 關鍵字。

>>> a.extend(b)
>>> a
[1, 2, 3, 4, 5, 6]

後兩種方式明顯更優雅,推薦使用。需要說明的一點是,如果清單很大的話, 會比較慢,使用 extend 更好。

清單元素去重

使用 set() 對清單元素進行去重。

>>> a = [1, 2, 3, 4, 2, 3]
>>> list(set(a))
[1, 2, 3, 4]

列表排序

使用 sort() 或內建函數 sorted() 對列表進行排序。它們之間的差異有兩點:

  • sort() 方法是對原始列表進行操作,而 sorted() 方法會傳回一個新列表,而不是在原來的基礎上操作。
  • sort() 是應用在清單上的方法,而 sorted() 可以對所有可迭代的物件進行排序運算。
# 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]

遍歷列表的索引和元素對

使用 enumerate() 函數可以同時輸出索引和元素值。

>>> a = ['python', 'go', 'java']
>>> for i, v in enumerate(a):
... print(i, v)

# output
0 python
1 go
2 java

找出清單中出現最頻繁的元素

使用 max() 函數可以快速找出一個清單中出現頻率最高的某個元素。

>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]
>>> b = max(set(a), key=a.count)
>>> b
4

需要說明的一點是,當清單中有兩個元素出現的次數相同時,會傳回第一個出現的元素。

>>> a = [1, 2]
>>> b = max(set(a), key=a.count)
>>> b
1

統計列表中所有元素的出現次數

前面的程式碼給出了出現最頻繁的值。如果想要知道清單中所有元素的出現次數,那麼可以使用 collections 模組。

collections 是 Python 中的一個寶藏模組,它提供了許多特性。 Counter 方法剛好可以完美解決這個需求。

>>> 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})

將兩個清單合併為字典

使用 zip() 函數,可以將兩個清單合併成字典。

>>> a = ['one', 'tow', 'three']
>>> b = [1, 2, 3]
>>> dict(zip(a, b))
{'one': 1, 'tow': 2, 'three': 3}

求兩個清單的交集,並集和差集。

# 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()

字典創建

# 1、创建空字典
a = {}
b = dict()

# 2、有初始值,从输入的便利程度来说,我更喜欢第二种
a = {'a': 1, 'b': 2, 'c': 3}
b = dict(a=1, b=2, c=3)

# 3、key 来自一个列表,而 value 相同, 使用 fromkeys,那是相当的优雅
keys = ['a', 'b', 'c']
value = 100
d = dict.fromkeys(keys, value)

# 4、key 来自一个列表,而 value 也是一个列表,使用 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 中判斷某個key 是否存在,可以使用has_key,但這個方法在Python3 中已經被移除了。

另一種方法是使用 in 關鍵字,不僅相容於 Python2 和 Python3,速度還更快,強烈建議。

d = {'a': 1, 'b': 2}
if 'a' in d:
print('hello')

取得字典中的值

d = {'a': 1, 'b': 2}

# 1、直接用 key 取值,但这种方式不好,如果 key 不存在会报错,推荐使用 get
a = d['a']

# 2、使用 get,如果 key 不存在还可以赋默认值
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

字典推導式

列表推導式和字典推導式是我相當喜歡的功能,簡潔高效。 map 和 filter 我都已經快不會用了。

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])

還有一個需求是我在開發過程經常碰到的,就是有一個列表,列表的元素是字典,然後按字典的value對清單進行排序。

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)

以上是我寫的 Python 程式碼,同事都說好的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:51cto.com。如有侵權,請聯絡admin@php.cn刪除