根据给定的长和宽,以及初始值,返回一个二维列表:
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]]
使用一个函数应用到一个数组的每个元素上,使得这个数组被切割成两个部分。如果说,函数应用到元素上返回的值为 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']]
两个数组在被一个函数应用后,从第一个数组中提取出共有的元素的原元素组成一个新的数组:
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]
返回数组中最大值的下标:
def max_element_index(arr): return arr.index(max(arr))
例如:
>>> max_element_index([5, 8, 9, 7, 10, 3, 0]) 4
找出两个数组中不同的元素,并合成为一个新的数组:
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]
如果 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
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}
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'
对列表中的各个字典里相同键值的对象求和:
def sum_by(lst, fn): return sum(map(fn,lst))
例如:
>>> sum_by([{ 'n': 4 }, { 'n': 2 }, { 'n': 8 }], lambda v : v['n']) 14
求出列表中某个数出现的次数和:
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
对一个列表根据所需要的大小进行细分:
from math import ceil def chunk(lst, size): return list ( map(lambda x:lst[x * size:x * size + size], list(range(0, int(ceil(len(lst) / size))))))
效果如下:
chunk([1,2,3,4,5],2) # [[1,2],[3,4],5]
return 中,map 的第二个参数是一个列表,map 会将列表中的每一个元素用于调用第一个参数的 function 函数,返回包含每次 function 函数返回值的新列表。
同样是一则关于 map 的应用,将整形数字拆分到数组中:
def digitize(n): return list(map(int, str(n)))
效果如下:
digitize(123) # [1, 2, 3]
它将整形数字n转化为字符串后,还自动对该字符串进行了序列化分割,最后将元素应用到 map 的第一个参数中,转化为整形后返回。
def fibonacci(n): if n <= 0: return [0] sequence = [0, 1] while len(sequence) <= n: next_value = (sequence[len(sequence) - 1] + sequence[len(sequence) - 2]) sequence.append(next_value) return sequence
效果如下:
fibonacci(7) # [0, 1, 1, 2, 3, 5, 8, 13]
⑭ 下划线化字符串 批量统一变量名称或者字符串格式:
from re import sub def snake(s): return '_'.join( sub('([A-Z][a-z]+)', r' 1', sub('( [A-Z]+)', r' l1', s.replace('-', ' '))).split()).lower()
效果如下:
snake('camelCase')# 'camel_case' snake('some text')# 'some_text' snake('some-mixed_string With spaces_underscores-and-hyphens')# 'some_mixed_string_with_spaces_underscores_and_hyphens' snake('AllThe-small Things')# "all_the_small_things"