python 生成list的所有的子集 (不使用递归且不引入标准库)
程序员文章站
2022-05-31 21:05:35
不使用递归且不引入标准库,单纯用两个for循环即可得出一个list的所有子集 L = [1, 2, 3, 4] List = [[]] for i in range(len(L)): # 定长 for j in range(len(List)): # 变长 sub_List = List[j] + ......
不使用递归且不引入标准库,单纯用两个for循环即可得出一个list的所有子集
l = [1, 2, 3, 4]
list = [[]]
for i in range(len(l)): # 定长
for j in range(len(list)): # 变长
sub_list = list[j] + [l[i]]
if sub_list not in l:
list.append(sub_list)
print('list =', list)
主要思想:
变长的 list 中的所有元素将会被不断地重复遍历,直到遍历完定长的 l
当然,不进行条件判断也行:
l = [1, 2, 3, 1] list = [[]] for i in range(len(l)): # 定长 for j in range(len(list)): # 变长 list.append(list[j] + [l[i]]) print('list =', list)
最后,如果考虑到程序的效率问题,那么建议引入 python 标准库中的 itertools,然后调用 combinations 这个函数
这样可以更加高效地得到一个 list 的所有的子集
代码如下:
from itertools import combinations l = [1, 2, 3, 1] result_list = sum([list(map(list, combinations(l, i))) for i in range(len(l) + 1)], []) print('result_list =', result_list)