Python列表生成式12個小功能,你常用哪幾個?

2021-09-29 13:40:43 字數 3521 閱讀 6357

作者 | zglg

** | python與演算法社群

python裡表示乙個列表,對容器型別的資料進行運算和操作,生成新的列表最高效、快速的辦法,就是列表生成式。

它優雅、簡潔,值得大家多多使用!今天盤點列表生成式在工作中的主要使用場景。

range快速生成連續列表

in [1]: a = range(11)

in [2]: a

out[2]: range(0, 11)

in [3]: list(a)

out[3]: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

對列表裡面的資料進行運算後重新生成乙個新的列表:

in [5]: a = range(0,11)

in [6]: b = [x**2 for x in a]

in [7]: b

out[7]: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

對乙個列表裡面的資料篩選,只計算[0,11) 中偶數的平方:

in [10]: a = range(11)

in [11]: c = [x**2 for x in a if x%2==0]

in [12]: c

out[12]: [0, 4, 16, 36, 64, 100]

前面列表生成式都只傳乙個引數x,帶有兩個引數的運算:

in [13]: a = range(5)

in [14]: b = ['a','b','c','d','e']

in [20]: c = [str(y) + str(x) for x, y in zip(a,b)]

in [21]: c

out[21]: ['a0', 'b1', 'c2', 'd3', 'e4']

結合字典,列印鍵值對:

in [22]: a = 

in [23]: b = [k+ '=' + v for k, v in a.items()]

in [24]: b = [k+ '=' + str(v) for k, v in a.items()]

in [25]: b

out[25]: ['a=1', 'b=2', 'c=3']

輸出某個目錄下的所有檔案和資料夾的名稱:

in [33]: [d for d in os.listdir('d:/summary')]

out[33]: ['a.txt.txt', 'python-100']

列表中所有單詞都轉化為小寫:

將值分組:

in [36]: def bifurcate(lst, filter):

...: return [

...: [x for i,x in enumerate(lst) if filter[i] == true],

...: [x for i,x in enumerate(lst) if filter[i] == false]

...: ]

...:

in [37]: bifurcate(['beep', 'boop', 'foo', 'bar'], [true, true, false, true])

out[37]: [['beep', 'boop', 'bar'], ['foo']]

進一步抽象例子8,根據指定函式fn 對lst 分組:

in [38]: def bifurcate_by(lst, fn):

...: return [

...: [x for x in lst if fn(x)],

...: [x for x in lst if not fn(x)]

...: ]

...:

in [39]: bifurcate_by(['beep', 'boop', 'foo', 'bar'], lambda x: x[0] == 'b')

out[39]: [['beep', 'boop', 'bar'], ['foo']]

返回可迭代物件的差集,注意首先都把a, b用set 包裝

in [53]: def difference(a, b):

...: _a, _b =set(a),set(b)

...: return [item for item in _a if item not in _b]

...:

...:

in [54]: difference([1,1,2,3,3], [1, 2, 4])

out[54]: [3]

進一步抽象10,根據函式fn 對映後選取差集,如下列表元素分別為單個元素和字典的例子:

in [61]: def difference_by(a, b, fn):

...: ...: _b = set(map(fn, b))

...: ...: return [item for item in a if fn(item) not in _b]

...: ...:

...:

in [62]: from math import floor

...: difference_by([2.1, 1.2], [2.3, 3.4],floor)

out[62]: [1.2]

in [63]: difference_by([, ], , lambda v : v['x'])

out[63]:

過濾非重複值,結合list 的count( 統計出元素在列表**現次數):

in [64]: def filter_non_unique(lst):

...: return [item for item in lst if lst.count(item) == 1]

in [65]: filter_non_unique([1, 2, 2, 3, 4, 4, 5])

out[65]: [1, 3, 5]

熟練操作以上12個例子,就算掌握python 中非常有用的列表生成式。

Python列表生成式12個小功能,你常用哪幾個?

作者 zglg python與演算法社群 python裡表示乙個列表,對容器型別的資料進行運算和操作,生成新的列表最高效 快速的辦法,就是列表生成式。它優雅 簡潔,值得大家多多使用!今天盤點列表生成式在工作中的主要使用場景。range快速生成連續列表 in 1 a range 11 in 2 a o...

Python 列表生成式

列表生成式即list comprehensions,是python內建的非常簡單卻強大的可以用來建立list的生成式。舉個例子,要生成list 1,2,3,4,5,6,7,8,9,10 可以用list range 1,11 list range 1,11 1,2,3,4,5,6,7,8,9,10 但...

python列表生成式

全都是重點!列表生成式即listcomprehensions,是python內建的非常簡單卻強大的可以用來建立list的生成式。舉個例子,要生成list 1,2,3,4,5,6,7,8,9,10 可以用range 1,11 range 1,11 1,2,3,4,5,6,7,8,9,10 但如果要生成...