1. 程式人生 > >numpy中takes函式

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.

官方例子