Python如何从列表中获取笛卡尔积

2024年06月09日 python列表 Python51

1、可以使用itertools.product在标准库中使用以获取笛卡尔积。

from itertools import product
 
somelists = [
   [1, 2, 3],
   ['a', 'b'],
   [4, 5]
]
 
result = list(product(*somelists))
print(result)

2、迭代方法。

def cartesian_iterative(pools):
  result = [[]]
  for pool in pools:
    result = [x+[y] for x in result for y in pool]
  return result

3、递归方法。

def cartesian_recursive(pools):
  if len(pools) > 2:
    pools[0] = product(pools[0], pools[1])
    del pools[1]
    return cartesian_recursive(pools)
  else:
    pools[0] = product(pools[0], pools[1])
    del pools[1]
    return pools
def product(x, y):
  return [xx + [yy] if isinstance(xx, list) else [xx] + [yy] for xx in x for yy in y]

4、Lambda方法。

def cartesian_reduct(pools):
  return reduce(lambda x,y: product(x,y) , pools)

本文链接:http://so.lmcjl.com/news/6226/

展开阅读全文