14个Python一行代码技巧：从二维列表到斐波那契数列

晚来风

发布时间：2026-10-14 | 4人浏览过

原创

本文精选14个Python高效短代码片段，涵盖二维列表初始化、数组分割、交集提取、最大值索引、对称差集、数值钳制、字典映射、大小写转换、键值求和、元素计数、列表分块、数字拆分、非递归斐波那契及字符串下划线化。每个技巧均提供核心代码与运行示例，帮助开发者快速掌握Python的高级数据处理能力。

① 二维列表

根据指定的宽度和高度，以及初始值，创建一个二维列表：


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 落在指定范围内，则返回该数值；否则返回距离该范围最近的边界值：


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)
# [[1,2],[3,4],5]

return 语句中，map 的第二个参数是一个列表，map 会将列表中的每个元素作为输入调用第一个参数的 function 函数，并返回包含每次函数返回值的新列表。

⑫ 数字转数组

同样利用 map 的应用，将整数拆分为数组元素：


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

效果如下：


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

该代码将整数 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)
# [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"

re.sub 用于替换字符串中的匹配项。这里涉及多层嵌套替换，初看可能不易理解，需逐步解析。
第一次替换，将字符串 s 中的 '-' 替换为 ' '（空格）。
第二次替换，针对第一次替换后的字符串，对符合 '([A-Z]+)' 正则表达式的字符区段（全大写单词）使用 r' \1' 进行替换，即用空格分隔每个单词。
第三次替换，针对第二次替换后的字符串，对符合 '([A-Z][a-z]+)' 正则表达式的字符区段（首字母大写、其余小写的词语）使用 r' \1' 进行替换，同样以空格分隔单词。

到此这篇关于Python之30秒就能学会的漂亮短程序代码的文章就介绍到这了,更多相关Python短程序代码内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站！