通过乘法从两个列表中创建新列表。 Python

Question

早上好！

我正在尝试使用乘法运算从两个列表中生成一个新列表。下面我一步一步地告诉你我做了什么：

import itertools
from itertools import product
import numpy as np
import pandas as pd

Parameter_list=[]

Parameter=[range(0,2,1),range(0,2,1)]
Parameter_list=list(itertools.product(*Parameter))
print(Parameter_list)

[(0, 0), (0, 1), (1, 0), (1, 1)]

然后我删除了第一个值，它基本上是空矩阵：

del Parameter_list[0]
print(Parameter_list)
[(0, 1), (1, 0), (1, 1)]

我接着创建了两个参数列表：

Parameter_A=[range(1,2,1),range(3,6,2),range(10,20,10)]
Parameter_A=list(itertools.product(*Parameter_A))

Parameter_B=[range(0,2,1),range(4,6,2),range(10,20,10)]
Parameter_B=list(itertools.product(*Parameter_B))

print(Parameter_A)
print(Parameter_B)

[(1, 3, 10), (1, 5, 10)]
[(0, 4, 10), (1, 4, 10)]

并合并列表：

comb=list(product(Parameter_A,Parameter_B))
print(comb)

[((1, 3, 10), (0, 4, 10)),
 ((1, 3, 10), (1, 4, 10)), 
 ((1, 5, 10), (0, 4, 10)), 
 ((1, 5, 10), (1, 4, 10))]

直到这里没有问题。但现在我正在努力通过将 参数列表 与 组合列表 相乘来创建一个新列表。所需的输出如下：

[((0, 0, 0), (0, 4, 10)),
 ((0, 0, 0), (1, 4, 10)), 
 ((0, 0, 0), (0, 4, 10)), 
 ((0, 0, 0), (1, 4, 10)),
 ((1, 3, 10), (0, 0, 0)),
 ((1, 3, 10), (0, 0, 0)), 
 ((1, 5, 10), (0, 0, 0)), 
 ((1, 5, 10), (0, 0, 0)),
 ((1, 3, 10), (0, 4, 10)),
 ((1, 3, 10), (1, 4, 10)), 
 ((1, 5, 10), (0, 4, 10)), 
 ((1, 5, 10), (1, 4, 10))]

有人可以帮助我吗？非常感谢！

Answer 1

使用列表而不是 numpy 数组来执行此操作并不是最方便的选择。就是说，这仍然是您可以使用一条线来完成的事情。

prod = [tuple(i if j != 0 else (0,) * len(i) for i, j in zip(comb_items, bool_items)) 
        for comb_items, bool_items in itertools.product(comb, Parameter_list)]

>>> prod
[((0, 0, 0),  (0, 4, 10)), 
 ((1, 3, 10), (0, 0, 0)), 
 ((1, 3, 10), (0, 4, 10)), 
 ((0, 0, 0),  (1, 4, 10)), 
 ((1, 3, 10), (0, 0, 0)), 
 ((1, 3, 10), (1, 4, 10)), 
 ((0, 0, 0),  (0, 4, 10)), 
 ((1, 5, 10), (0, 0, 0)), 
 ((1, 5, 10), (0, 4, 10)), 
 ((0, 0, 0),  (1, 4, 10)), 
 ((1, 5, 10), (0, 0, 0)), 
 ((1, 5, 10), (1, 4, 10))]

我假设输出的顺序并不重要并且 Parameter_list 将始终是布尔值。如果需要，这两件事都可以很容易地改变。

通过乘法从两个列表中创建新列表。 Python

Create new list out of two lists via multiplication. Python

python

cartesian-product

itertools

matrix-multiplication