今天给大家带来一些30秒就能学会的代码片段,这些代码潜力无限,蕴含了丰富的python编程思维,应用领域非常广泛,而且学起来非常简单。
1. 二维列表
解读:根据给定的长和宽,以及初始值,返回一个二维列表。
def initialize_2d_list(w, h, val=None):
return [[val for x in range(w)] for y in range(h)]
例:
>>> 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)]
]
例:
>>> 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]
例:
>>> 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))
例:
>>> 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]
例:
>>> 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))
例:
>> 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
例:
>>> 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:])
例:
>>> decapitalize('FooBar')
'fooBar'
>>> decapitalize('FooBar', True)
'fOOBAR'
9. 同键求和
解读:对列表中的各个字典里相同键值的对象求和。
def sum_by(lst, fn):
return sum(map(fn,lst))
例:
>>> 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)])
例:
>>> count_occurrences([1, 1, 2, 1, 2, 3], 1)
3
11.数组再分组
对一个列表根据所需要的大小进行细分:
效果如下: