numpy中takes函数

numpy.take(a, indices, axis=None, out=None, mode='raise'）

take(indices[, axis, out, mode]) :提取指定索引位置的数据,并以一维数组或者矩阵返回(主要取决axis)

>>> a = [4, 3, 5, 7, 6, 8]
>>> indices = [0, 1, 4]
>>> np.take(a, indices)
array([4, 3, 6])

pandas中的

DataFrame.take(indices, axis=0, convert=None, is_copy=True, **kwargs)[source]
Return the elements in the given positional indices along an axis.
##参数含义
indices : array-like
# An array of ints indicating which positions to take.

axis : {0 or ‘index’, 1 or ‘columns’, None}, default 0
# The axis on which to select elements. 0 means that we are selecting rows, 1 means that we are selecting columns.

官方例子