Python生成给定列表的笛卡尔积

文章目录

  • 一,背景
  • 二、函数介绍
  • 三、示例

一,背景

data = [["165465","dhfh-中国,每列汇总嗯嗯好","春,夏,秋","男,女,中性","","","美丽,瘦身","",""],
        ["165465","dhfh-中国,每列汇总嗯嗯好","春","男","","","美丽","chehis","cehsi"]]

将上面的数据拆分为下面这种形式

data = [['45456', '名字', '男', '', '', '', '美丽'], ['45456', '名字', '男', '', '', '', '瘦身']]

二、函数介绍

itertools.product函数是Python标准库中的一个函数,位于itertools模块中。它用于生成多个可迭代对象的笛卡尔积。

函数签名为:itertools.product(*iterables, repeat=1)

参数解释:

*iterables:表示接受多个可迭代对象作为参数,这些对象将用来生成笛卡尔积。每个可迭代对象代表输入集合的一个维度。
repeat:可选参数,表示重复生成的次数。
返回值:一个迭代器,该迭代器生成所有可能的组合。

下面是一个更详细的示例,展示了itertools.product函数的使用方法:

import itertools

# 定义两个可迭代对象
letters = ['a', 'b']
numbers = [1, 2]

# 调用itertools.product函数并传入两个可迭代对象
product_result = itertools.product(letters, numbers)

# 使用for循环遍历迭代器并打印结果
for item in product_result:
    print(item)

输出结果将会是:

('a', 1)
('a', 2)
('b', 1)
('b', 2)

在这个例子中,我们定义了两个可迭代对象lettersnumbers,分别表示字母和数字。然后,我们调用itertools.product函数,并传入这两个可迭代对象作为参数。

product_result是一个迭代器,它生成了两个可迭代对象的笛卡尔积的所有可能组合。最后,我们使用for循环遍历这个迭代器,并打印每个组合。

三、示例


import itertools
from pprint import pprint


data = [["165465","dhfh-中国,每列汇总嗯嗯好","春,夏,秋","男,女,中性","","","美丽,瘦身","",""],
        ["165465","dhfh-中国,每列汇总嗯嗯好","春","男","","","美丽","chehis","cehsi"]]

b = []
for i in data:
    a = [item.split(",") for item in i]
    result = [list(combination) for combination in itertools.product(*a)]
    b+=result

pprint(b)
print(len(b))

你可能感兴趣的:(Python,(学习使我快乐),python,开发语言)