你可以使用itertools模块中的product函数来计算保留每个项索引位置的数组的笛卡尔积。以下是一个使用代码示例:
import itertools
def cartesian_product(arrays):
# 计算每个数组的长度
lengths = [len(array) for array in arrays]
# 创建每个数组的索引列表
indices = [range(length) for length in lengths]
# 使用itertools的product函数计算笛卡尔积
product = list(itertools.product(*indices))
# 根据索引列表获取每个项的值
result = [[array[index] for array, index in zip(arrays, indices)] for indices in product]
return result
# 测试代码
arrays = [[1, 2, 3], ['a', 'b', 'c'], [True, False]]
result = cartesian_product(arrays)
print(result)
输出结果为:
[[1, 'a', True], [1, 'a', False], [1, 'b', True], [1, 'b', False], [1, 'c', True], [1, 'c', False], [2, 'a', True], [2, 'a', False], [2, 'b', True], [2, 'b', False], [2, 'c', True], [2, 'c', False], [3, 'a', True], [3, 'a', False], [3, 'b', True], [3, 'b', False], [3, 'c', True], [3, 'c', False]]
这个代码示例中,我们使用了itertools模块中的product函数来计算笛卡尔积。首先,我们计算了每个数组的长度,并创建了每个数组的索引列表。然后,我们使用product函数计算了索引列表的笛卡尔积。最后,我们根据索引列表获取了每个项的值,得到了最终的结果。
上一篇:保留每个唯一前缀中的一个项目。
下一篇:保留每个用户最近的X个条目