0
鉴于形状(5,3)
的numpy的矩阵a
和索引向量形状(5,)
的b
,其中在索引向量中的每个条目是0之间〜2,如何可以创建基于a
新向量c
及其索引向量b
。如何索引这个numpy数组?
鉴于形状(5,3)
的numpy的矩阵a
和索引向量形状(5,)
的b
,其中在索引向量中的每个条目是0之间〜2,如何可以创建基于a
新向量c
及其索引向量b
。如何索引这个numpy数组?
为其他尺寸使用arange
:
c = a[np.arange(5), b]