2017-03-06 52 views
0

鉴于形状(5,3)的numpy的矩阵a和索引向量形状(5,)b,其中在索引向量中的每个条目是0之间〜2,如何可以创建基于a新向量c及其索引向量b如何索引这个numpy数组?

回答

1

为其他尺寸使用arange

c = a[np.arange(5), b]