将整数数组编码为唯一的 int

我有固定数量的 int 数组，其形式为：

[a,b,c,d,e]

例如：

[2,2,1,1,2]

where a and b可以是 0 到 2 之间的整数，c and d可以是 0 或 1，并且e可以是 0 到 2 之间的整数。

因此有：3 * 3 * 2 * 2 * 3: 108这种形式的可能数组。

我想为每个数组分配一个从 0 到 107 的唯一整数代码。

我被困住了，我想将数组中的每个数字相加，但是两个数组，例如：

[0,0,0,0,1] and [1,0,0,0,0]

都会加到 1。

有什么建议吗？

谢谢。

你可以使用np.ravel_multi_index https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.ravel_multi_index.html:

>>> np.ravel_multi_index([1, 2, 0, 1, 2], (3, 3, 2, 2, 3))
65

验证：

>>> {np.ravel_multi_index(j, (3, 3, 2, 2, 3)) for j in itertools.product(*map(range, (3,3,2,2,3)))} == set(range(np.prod((3, 3, 2, 2, 3))))
True

从另一个方向回去：

>>> np.unravel_index(65, dims=(3, 3, 2, 2, 3))
(1, 2, 0, 1, 2)