Python 自动化

Question

下面的问题可能有点令人困惑，但我会尽力以最好的方式解释它。

假设我们为一家制造公司工作。它制造然后用于制造产品的组件。在以下示例中，有 3 个组件和 2 个最终产品。

组件和产品的需求如下:

comp1 = 8
comp2 = 3
comp3 = 5
prod1 = 2
prod2 = 2

可以保存在字典中的:

clst = {"comp": [comp1, comp2, comp3], "prod": [prod1, prod2]}

另一方面，要制造最终产品，组件要求如下:

prod1 需要:2 个 comp1、1 个 comp2、1 个 comp3

产品 2 需要:3 个组件 1、1 个组件 2、2 个组件 3

在字典中:

rprod = {0: [2,1,1], 1: [3,1,2]}

我需要创建一个列表，其中包含按以下方式混合数据的 Material 贡献和消耗列表:

[
    [
        [1, 1, 1, 1, 1, 1, 1, 1, -2],            # ---->comp1/prod1(1)
                 [1, 1, 1, -1],               # ---->comp2/prod1(1)
                [1, 1, 1, 1, 1, -1]         # ---->comp3/prod1(1)
    ]
    ,
    [
        [1, 1, 1, 1, 1, 1, 1, 1, -2],            # ---->comp1/prod1(2)
                 [1, 1, 1, -1],               # ---->comp2/prod1(2)
                [1, 1, 1, 1, 1, -1]         # ---->comp3/prod1(2)
    ]
    ,
    [
        [1, 1, 1, 1, 1, 1, 1, 1, -3],            # ---->comp1/prod2(1)
                 [1, 1, 1, -1],               # ---->comp2/prod2(1)
                [1, 1, 1, 1, 1, -2]         # ---->comp3/prod3(1)
    ]
    ,
    [
        [1, 1, 1, 1, 1, 1, 1, 1, -3],            # ---->comp1/prod2(2)
                 [1, 1, 1, -1],               # ---->comp2/prod2(2)
                [1, 1, 1, 1, 1, -2]         # ---->comp3/prod3(2)
    ]
]

列表中1的数量对应该组件的需求数量，最后一个负数为该产​​品的组件需求。

对制作 list 有什么建议吗？

Answer 1

In[2]: comp1 = 8
  ...: comp2 = 3
  ...: comp3 = 5
  ...: prod1 = 2
  ...: prod2 = 2
  ...: 
  ...: clst = {"comp": [comp1, comp2, comp3], "prod": [prod1, prod2]}
  ...: rprod = {0: [2,1,1], 1: [3,1,2]}
In[3]: result = []
  ...: for i, p in enumerate(clst['prod']):
  ...:     for _ in range(p):
  ...:         tmp = [([1] * a) + [-b] for a, b in zip(clst['comp'], rprod[i])]
  ...:         result.append(tmp)
  ...: 
In[4]: result
Out[4]: 
[[[1, 1, 1, 1, 1, 1, 1, 1, -2], [1, 1, 1, -1], [1, 1, 1, 1, 1, -1]],
 [[1, 1, 1, 1, 1, 1, 1, 1, -2], [1, 1, 1, -1], [1, 1, 1, 1, 1, -1]],
 [[1, 1, 1, 1, 1, 1, 1, 1, -3], [1, 1, 1, -1], [1, 1, 1, 1, 1, -2]],
 [[1, 1, 1, 1, 1, 1, 1, 1, -3], [1, 1, 1, -1], [1, 1, 1, 1, 1, -2]]]