欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python 生成list的所有的子集 (不使用递归且不引入标准库)

程序员文章站 2023-11-14 17:43:16
不使用递归且不引入标准库,单纯用两个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)