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]