wolfram-mathematica - 将 1D 列表转换为 3D 列表

Question

我有将 1D 列表转换为 3D 列表的函数，但同时当 2 维和 3 维索引相等时，它会放置零而不是输入列表中的值:

n = 4

input = Table[RandomInteger[5], {i, 1, 48}]

convert[l_] := Table[If[i == j, 0, l[[index++]]], {s, 1, 4}, {i, 1, n}, {j, 1, n}]

output = convert[input]

我想去掉 Increment[] (++) 函数。

Answer 1

我认为这可能有效:

convert[l_] := Insert[Partition[Partition[l, 3], 4], 0, 
                      Flatten[Table[{j, i, i}, {j, 4}, {i, 4}], 1]]

也许更精明的人可以帮助摆脱 Partition[Partition 令人讨厌的构造。

喂!