2017-03-06 50 views
0

鑑於形狀(5,3)的numpy的矩陣a和索引向量形狀(5,)b,其中在索引向量中的每個條目是0之間〜2,如何可以創建基於a新向量c及其索引向量b如何索引這個numpy數組?

回答

1

爲其他尺寸使用arange

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