仅用30秒，Python就能学会的漂亮短代码，你学会了吗？

🤵‍♂️ 个人主页@老虎也淘气 个人主页
✍🏻作者简介：Python学习者
🐋 希望大家多多支持我们一起进步！😄
如果文章对你有帮助的话，
欢迎评论 💬点赞👍🏻 收藏 📂加关注

1.“二维列表”

**解读：**根据给定的长和宽，以及初始值，返回一个二维列表。

def initialize_2d_list(w, h, val=None):
    return [[val for x in range(w)] for y in range(h)]

eg

>>> initialize_2d_list(2,2)
[[None, None], [None, None]]

>>> initialize_2d_list(2,2,0)
[[0, 0], [0, 0]]

2.函数切割数组

**解读：**使用一个函数应用到一个数组的每个元素上，使得这个数组被切割成两个部分。如果说，函数应用到元素上返回的值为True，则该元素被切割到第一部分，否则分为第二部分。

def bifurcate_by(lst, fn):
    return [
      [x for x in lst if fn(x)],
      [x for x in lst if not fn(x)]
    ]

eg

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

3.“交集点”

解读： 两个数组在被一个函数应用后，从第一个数组中提取出共有的元素的原元素组成一个新的数组。

def intersection_by(a, b, fn):
    _b = set(map(fn, b))
    return [item for item in a if fn(item) in _b]

eg

>>> from math import floor
>>> intersection_by([2.1, 1.2], [2.3, 3.4],floor)
[2.1]

4.最大值下标

**解读：**返回数组中最大值的下标。

def max_element_index(arr):
    return arr.index(max(arr))

eg

>>> max_element_index([5, 8, 9, 7, 10, 3, 0])
4

5.数组对称差

**解读：**找出两个数组中不同的元素，并合成为一个新的数组。

def symmetric_difference(a, b):
    _a, _b = set(a), set(b)
    return [item for item in a if item not in _b] + [item for item in b if item not in _a]

eg

>>> symmetric_difference([1, 2, 3], [1, 2, 4])
[3, 4]

6.“夹数”

**解读：**如果 num 落在一段数字范围内，则返回num，否则返回离这个范围最近的边界：

def clamp_number(num,a,b):
    return max(min(num, max(a,b)),min(a,b))

eg

>> clamp_number(2,3,10)
3

>> clamp_number(7,3,10)
7

>> clamp_number(124,3,10)
10

7.键值映射

**解读：**使用对象的键重新创建对象，并运行函数为每个对象的键创建值。
使用dict.keys()遍历对象的键， 通过函数生成一个新的值。

def map_values(obj, fn):
    ret = {}
    for key in obj.keys():
        ret[key] = fn(obj[key])
    return ret

eg

>>> users = {
...   'fred': { 'user': 'fred', 'age': 40 },
...   'pebbles': { 'user': 'pebbles', 'age': 1 }
... }

>>> map_values(users, lambda u : u['age'])
{'fred': 40, 'pebbles': 1}

>>> map_values(users, lambda u : u['age']+1)
{'fred': 41, 'pebbles': 2}

8.大小写转换

解读： 将英文单词的首字母大写改为小写。
upper_rest参数：设定是否将除首字母外的其他字母大小写转换。

def decapitalize(s, upper_rest=False):
    return s[:1].lower() + (s[1:].upper() if upper_rest else s[1:])

eg

>>> decapitalize('FooBar')
'fooBar'

>>> decapitalize('FooBar', True)
'fOOBAR'

9.同键求和

**解读：**对列表中的各个字典里相同键值的对象求和。

def sum_by(lst, fn):
    return sum(map(fn,lst))

eg

>>> sum_by([{ 'n': 4 }, { 'n': 2 }, { 'n': 8 }], lambda v : v['n'])
14

10.一行代码求出现次数

**解读：**求出列表中某个数出现的次数和。

def count_occurrences(lst, val):
    return len([x for x in lst if x == val and type(x) == type(val)])

eg

>>> count_occurrences([1, 1, 2, 1, 2, 3], 1)
3

11.数字转数组

同样是一则关于map的应用，将整形数字拆分到数组中：

def digitize(n):
    return list(map(int, str(n)))

效果如下：

digitize(123)
# [1, 2, 3]