如何在numpy中解释此数组索引？

Question

我想在以下代码片段中解释数组索引。 State[t,Con]是什么意思， Con本身是一个数组吗？

for t in range(T):  # 0 .. T-1
    State[t+1] = Bool[:, sum(Pow * State[t,Con],1)].diagonal()

Con如下所示（其中N> K）：

Con = apply_along_axis(random.permutation, 1, tile(range(N), (N,1) ))[:, 0:K]

Answer 1

Con是(N,K)个整数数组。

State大概是(T,N)数组。

State[t,Con]将是一个(N,K)从所选择的值的阵列t的行State 。 由于Con具有重复，因此State行的某些值将重复。

`Bool[:, sum(Pow * State[t,Con],1)].diagonal()`

然后，它与Pow （也是（N，K）数组，或兼容的东西）逐个元素相乘。 然后在最后一个轴（列）上求和，得到一个N,)数组（N个元素向量）。 然后从Bool数组（ (N,N)数组？）中选择那些列。 最终得到主对角线-再次为N值。

最后一步是将这些值分配给State的t+1行。