① 二维列表
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
⑦ 键值映射
使用对象的键重新创建对象,并运行函数为每个对象的键创建值; 使用 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 }
⑧ 大小写转换
将英文单词的首字母大写改为小写; 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'
⑨ 同键求和
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 )
return 中,map 的第二个参数是一个列表,map 会将列表中的每一个元素用于调用第一个参数的 function 函数,返回包含每次 function 函数返回值的新列表。
⑫ 数字转数组
同样是一则关于 map 的应用,将整形数字拆分到数组中:
def digitize ( n) :
return list ( map ( int , str ( n) ) )
digitize( 123 )
它将整形数字n转化为字符串后,还自动对该字符串进行了序列化分割,最后将元素应用到 map 的第一个参数中,转化为整形后返回。
⑬ 非递归斐波那契
菲波那切数列,前两个数的和为第三个数的值,如0、1、1、2、3、5、8、13… 如果使用递归来实现这个算法,效率非常低下,我们使用非递归的方式实现:
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 )
⑭ 下划线化字符串
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' )
snake( 'some text' )
snake( 'some-mixed_string With spaces_underscores-and-hyphens' )
snake( 'AllThe-small Things' )
re.sub 用于替换字符串中的匹配项。这里其实是一个“套娃”用法,一开始可能不太好理解,需要慢慢理解。 第一个替换,是将 s 字符串中,使用 ’ ’ 替换 ‘-’。 第二个替换,是针对第一个替换后的字符串,对符合 ‘([A-Z]+)’ 正则表达式的字符区段(全大写的单词)用 r’ \1’ 替换,也就是用空格区分开每一个单词。 第三个替换,是对第二个替换后的字符串,对符合 ‘([A-Z][a-z]+)’ 正则表达式的字符区段(也就是首字母大写,其他字母小写的词语)用 r’ \1’ 替换,也是将单词用空格分隔开。