numpy.take()用法

2023-09-04 17:21| 来源: 网络整理| 查看: 265

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])], [6, 9, 4, ([3, 7, 5])]]) print(a.take(1,0)) print(a) [out] [3 2 6 list([6, 5, 1])] [[1 2 4 list([1, 2, 5])] [3 2 6 list([6, 5, 1])] [6 9 4 list([3, 7, 5])]] a = np.array([[1, 2, 4, ([1, 2, 5])], [3, 2, 6, ([6, 5, 1])], [6, 9, 4, ([3, 7, 5])]]) print(a.take(1,1)) print(a) [out] [2 2 9] [[1 2 4 list([1, 2, 5])] [3 2 6 list([6, 5, 1])] [6 9 4 list([3, 7, 5])]]

【本文地址】

公司简介

联系我们