numpy的数组嵌套操作

numpy的ndarray[ndarray] 嵌套操作

直接demo

numpy的数组嵌套操作

看下面示意图,红线代表a[b]中的数据元素来自a中的相应位置,这个位置是由b数组中的元素来指出

numpy的数组嵌套操作

总结: a[b] 生成一个numpy的数组c,c的shape=(b.shape[0],b.shape[1],a.shape[1])。c的中的数据元素都是由a的数据元素组成,至于是a中那个位置的数据元素,则是由b中的数值指出,即b是a的索引