文章目录
列表展开
降维
分块
转置
查找众数
判断重复元素
比较元素差异
比较列表计算后的差异
为列表添加索引
回文
列表展开
对于形如[1, [2], [[3], 4], 5]的列表，将其展开为[1,2,3,4,5]。算法思路为，遍历列表A0中的元素，如果该元素A1仍为列表，则将A1的元素追加到A0的末尾。
def flatten(lst):
i = 0
while(i<len(lst)):
if isinstance(lst[i],list):
lst.extend(lst[i])
i += 1
return [n for n in lst if not isinstance(n,list)]
>>> lst = [1, [2], [[3], 4], 5]
>>> flatten(lst)
[1, 5, 2, 4, 3]
降维
当列表中的所有数据均为列表时，可以通过单星号索引之后，用itertools中的chain函数进行拼接。
>>> from itertools import chain
>>> a = [[1, 2], [3, 4], [5, 6]]
>>> chain(*a)
[1, 2, 3, 4, 5, 6]
分块
python内置了map函数，可以遍历列表并对每个元素进行映射。对于list(map(func,myList))，等价于[func(e) for e in myList]。根据这个特性，可以对列表进行分块。
from math import ceil #ceil为向上取整
def chunkByFor(lst, size):
return [lst[x*size:x*size+size]
for x in range(0,ceil(len(lst)/size))]
def chunkByMap(lst, size):
return list(
map(lambda x: lst[x * size:x * size + size],
list(range(0, ceil(len(lst) / size)))))
chunk([1,2,3,4,5],2) # [[1,2],[3,4],5]
转置
zip可以像拉链一样将数组中对应的值缝合起来，以元组的形式重新存储。根据这个特性，可完成列表的\”转置\”。
>>> lst = [[1,2], [3,4], [5,6]]
>>> list(zip(*lst))
[(1, 3, 5), (2, 4, 6)]
查找众数
list封装了count方法，可以统计其中的元素个数，例如
>>> lst = [1,2,1,2,3,2,1,4,2]
>>> lst.count(1)
3 #即lst中1出现了3次
而最大值选取函数max中有一个参数key，表示选取最大值的依据，例如
>>> x = [-2,-1,0,1,2]
>>> def test(x): return x**2-2*x
…
>>> max(x,key=test)
-2
表示选出当函数test值最大时的自变量。
根据这两个特点，可以得到查找列表中出现次数最多元素的方法为
>>> lst = [1,2,3,4,5,2,3,2,1,2,4]
>>> max(set(lst),key=lst.count)
2
判断重复元素
借助set中元素不重复的特性，可以检验非set数据中是否有重复数据。
def isUnique(lst):
return len(lst) == len(set(lst))
>>>x = [1,2,3,4,5,5]
>>>y = [1,2,3,4,5]
>>>isUnique(x)
False
>>>isUnique(y)
True
比较元素差异
在python中，集合set提供了类似做差的操作difference。对于集合A和B而言，若想实现A-B的功能，可采取A.difference(B)的方案。
对于其他数据结构，可先转为set再行做差。且difference会自动将输入变量转为set。
>>> A = [1,2,3]
>>> B = [1,2,4]
>>> list(set(A).difference(B))
[3]
比较列表计算后的差异
现有一函数func,对于两组数据A和B，如果希望得到func(A)和func(B)的差别，也可以采用set的特性。
def funcMinus(A, B, func):
B = set(map(func, B)) #剔除func(B)中的重复元素
return [a for a in A if fn(a) not in B]
筛选数据
filter是python的内置函数，用于过滤不符合条件的数据。list(filter(func,myList))等价于[e for e in myList if func(e)==1]。
为列表添加索引
enumerate可以将列表、元组等组合为索引序列，并可以声明初始值，例如
>>> orders = [\’first\’,\’second\’,\’third\’]
>>> list(enumerate(orders))
[(0, \’first\’), (1, \’second\’), (2, \’third\’)]
>>> list(enumerate(orders,2))
[(2, \’first\’), (3, \’second\’), (4, \’third\’)]
>>>
这可以非常方便地应用到for循环中
>>> for i,val in enumerate(orders,1):
… print(i,val)
…
1 first
2 second
3 third
回文
在python中，对列表或者字符串采用:进行索引，例如a:b指的是从a到b的数据；当采用双引号::时，引号间的值的意义就发生了变化，例如a:b:c表示从a到b，间隔为c的数据。
据此，可以得到::-1表示将字符串颠倒过来，据此可以判断一个字符串是否为回文结构
>>> s = \”abcde\”
>>> s[::-1]
\’edcba\’
>>> s == s[::-1]
False
版权声明：本文为CSDN博主「微小冷」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/m0_37816922/article/details/128756170
