python - 一维索引到多维索引的张量扩展

Question

我想从具有任何维度的一维索引数组中创建一个多维数组的函数。例如，我有

x=np.array([1,2,3])
def extend_array(x,d)

如果我把 d=2，那么结果数组是

[[1,1],
 [1,2],
 [1,3],
 [2,1],
 [2,2],
 [2,3],
 [3,1],
 [3,2],
 [3,3]]

或输入可以是 (n,d) 而不是 (x,d) 以便 n 指

np.array(range(1,n+1))

我认为有一个功能，但我找不到。你能帮我解决这个问题吗？

标签： pythonarraysindexing

Answer 1

from itertools import product
t=list(product(x,repeat=d))

这将给出预期的结果。