numpy.take()用法
学了Python很长时间了但是一直没有去研究take的使用方法
今天周末加班研究一下
写一个简单的笔记,基本上都是借鉴别人的
其旨在根据提供的索引值将元素形成数组输出,如下图所示;
numpy.take(a,indices,axis = None,out = None,mode =‘raise’ )
沿轴取数组中的元素。
当axis不是None时,此函数与“fancy”索引(使用数组索引数组)的功能相同; 但是,如果您需要沿给定轴的元素,则可以更容易使用。这样的呼叫 相当于 。np.take(arr, indices, axis=3)arr[:,:,:,indices,…]
如果a是一个numpy array,a.take(m,1)表示取每一行的第m个值;a.take(m,0)表示取第m行
a = np.array([[1, 2, 4, ([1, 2, 5])],
[3, 2, 6, ([6, 5, 1])