Python Numpy数组Mult组合

Question

我有一个1乘N和一个N乘1数组。在

至于Lin，我们都知道，当这两个相乘时，我们得到的是1乘1或一个值。在

N是我们想要的值。但是我想随机生成第一个数组，而第二个数组有权重。这是一个运筹学问题。在

[第0随机，第1随机，N-1随机]*[[0]，[1]，[2]，…，[N-1]]

我的代码是这样的：

import itertools
import numpy as np
N = 4
winner = 0
A = np.array(list(itertools.product(range(N-1), repeat = N-1)))
V_weights = np.arange(1, N, 1).reshape(N-1,1)
for array in A:
    if array.dot(V_weights) == N:
        # print(array)
        # print(V_weights)
        # print(array.dot(V_weights))
        winner += 1
print(winner)

我希望while循环在所有组合完成后结束。理想情况下是正确的迭代量。在

例如：我有4种原木，1磅，2磅，3磅，4磅，需要的木材总量（比如10磅）

这就给了我几种木材类型的组合：

10 1磅 3磅，2磅，13磅 阿尔索 2磅，13磅，15磅

等等

因此，我想我会生成随机数组，直到我得到所有的操作。在

抱歉，如果这有点混乱-很高兴编辑，除非有人需要一个更快更简单的方法。在

Answer 1

请看一下对choice的调用集：

In [27]: np.random.choice(range(4), size=(1,3), replace=False)
Out[27]: array([[1, 2, 3]])
In [28]: np.random.choice(range(4), size=(1,3), replace=False)
Out[28]: array([[2, 1, 3]])
In [29]: np.random.choice(range(4), size=(1,3), replace=False)
Out[29]: array([[1, 2, 0]])
In [30]: np.random.choice(range(4), size=(1,3), replace=False)
Out[30]: array([[1, 3, 2]])
In [31]: np.random.choice(range(4), size=(1,3), replace=False)
Out[31]: array([[2, 3, 0]])
In [32]: np.random.choice(range(4), size=(1,3), replace=False)
Out[32]: array([[1, 3, 2]])

注意[1,3,2]出现了两次。在

这将从这个范围产生所有长度为3的排列（不是以随机顺序排列-尽管列表可以被无序排列）

^{pr2}$

这些排列的加权值：

In [38]: wgts = np.arange(1, 4)
In [39]: A = np.array(list(itertools.permutations(range(4),3)))
In [40]: A.shape
Out[40]: (24, 3)
In [41]: wgts
Out[41]: array([1, 2, 3])
In [42]: A.dot(wgts)
Out[42]: 
array([ 8, 11,  7, 13,  9, 12,  7, 10,  5, 14,  7, 13,  5, 11,  4, 13,  8,
       11,  6,  9,  5, 11,  7, 10])

---

根据你的评论，我意识到你正在用替代品抽样。这里有一个快速的暴力方式来做这样的搜索

生成一组测试值。由于这是对每一行进行替换，因此我们只需调用一次choice，就可以生成许多行：

In [55]: A = np.random.choice(range(4), size=(100,3), replace=True)
In [56]: A
Out[56]: 
array([[3, 1, 1],
       [0, 3, 0],
       [1, 3, 2],
       [3, 1, 1],
       [2, 0, 2],
       [0, 3, 2],
       ...
       [3, 0, 3]])

如前所述，取加权和：

In [57]: wgts = np.arange(1, 4)
In [58]: A.dot(wgts)
Out[58]: 
array([ 8,  6, 13,  8,  8, 12,  5, 10,  9, 12,  9,  9,  7,  9,  0,  4,  3,
        6, 10,  8,  7,  2,  3, 14, 11,  7,  1,  7,  2,  5, 11,  3, 13, 12,
        7,  9,  4, 15,  1,  7,  7,  9, 10,  7,  9, 11, 14,  3, 13, 11,  2,
        9,  9,  2, 11, 16,  4, 10, 13,  9, 11,  8, 10,  5,  7,  8, 13, 15,
       11,  9, 13,  7,  6,  9,  5, 12,  9, 11,  6,  3,  0,  0,  9,  7, 11,
        0, 12,  9,  7,  7,  3,  7, 13,  3,  9,  5, 14,  1, 16, 12])

并找出该总和具有目标值的指数：

In [59]: np.where(_==4)
Out[59]: (array([15, 36, 56], dtype=int32),)
In [60]: A[_]
Out[60]: 
array([[2, 1, 0],
       [2, 1, 0],
       [0, 2, 0]])

对于N=5：

In [61]: A = np.random.choice(range(5), size=(100,4), replace=True)
In [62]: A.dot(np.arange(1,5))
Out[62]: 
array([29, 20,... 23])
In [63]: np.where(_==5)
Out[63]: (array([], dtype=int32),)

这个样本不够大，让我们试试大一点的：

^{8}$

---

itertools.product可用于通用所有组合

In [71]: A = np.array(list(itertools.product(range(4),repeat=3)))
In [72]: A.shape
Out[72]: (64, 3)
In [73]: x = A.dot(np.arange(1,4))
In [74]: A[x==4]
Out[74]: 
array([[0, 2, 0],
       [1, 0, 1],
       [2, 1, 0]])

N=5时：

In [75]: A = np.array(list(itertools.product(range(5),repeat=4)))
In [76]: A.shape
Out[76]: (625, 4)
In [77]: x = A.dot(np.arange(1,5))
In [78]: A[x==5]
Out[78]: 
array([[0, 1, 1, 0],
       [1, 0, 0, 1],
       [1, 2, 0, 0],
       [2, 0, 1, 0],
       [3, 1, 0, 0]])

---

为了迭代地进行搜索，我将使用product作为生成器：

In [100]: g = itertools.product(range(5), repeat=4)
In [101]: for cnt, row in enumerate(g):
     ...:     if np.arange(1,5).dot(row)==5:
     ...:         print(cnt, row)
     ...: 
     ...:     
30 (0, 1, 1, 0)
126 (1, 0, 0, 1)
175 (1, 2, 0, 0)
255 (2, 0, 1, 0)
400 (3, 1, 0, 0)

或者只找到第一个：

In [102]: g = itertools.product(range(5), repeat=4)
In [103]: for cnt, row in enumerate(g):
     ...:     if np.arange(1,5).dot(row)==5:
     ...:         print(cnt, row)
     ...:         break
     ...:     
30 (0, 1, 1, 0)