python必备小技巧

检验一个列表中的所有元素是否一样

使用[1:]和[:-1]来比较给定列表的所有元素

def all_equal(lst):
  return lst[1:] == lst[:-1]
  
all_equal([1, 2, 3, 4, 5, 6]) # False
all_equal([1, 1, 1, 1]) # True

如果列表所有值都是唯一的，返回 True，否则 False

在给定列表上使用集合set()去重，比较它和原列表的长度

def all_unique(lst):
  return len(lst) == len(set(lst))

x = [1,2,3,4,5,6]
y = [1,2,2,3,4,5]
all_unique(x) # True
all_unique(y) # False

返回两个iterables间的差异

创建b的集合，使用a的列表推导式保留不在_b中的元素

def difference(a, b):
  _b = set(b)
  return [item for item in a if item not in _b]

difference([1, 2, 3], [1, 2, 2, 4]) # [3]

一次性整合列表

使用嵌套的列表提取字列表的每个值

def merge(lst):
	return [x for y in lst for x in y]
merge([[1,2,3,4,5],[5,4,3,2,1]])

将一个数分解转换为个位数字

将n字符化后使用map()函数结合int完成转化

def digitize(n):
  return list(map(int, str(n)))
  
digitize(123) # [1, 2, 3]