mars.tensor.delete#

mars.tensor.delete(arr, obj, axis=None)[来源]#

返回一个新数组，沿着一个轴删除子数组。对于一维数组，这返回那些没有被arr[obj]返回的条目。

Parameters

arr (类数组) – 输入数组。
obj (slice, int 或 array 的 ints) – 指示要沿指定轴移除的子数组的索引。
axis (int, 可选) – 沿着删除由 obj 定义的子数组的轴。如果 axis 为 None，则 obj 应用于扁平化数组。

Returns

out – 一个复制的 arr，其中包含由 obj 指定的元素被移除。请注意 delete 并不会就地发生。如果 axis 为 None，out 将是一个扁平化数组。

Return type

mars.tensor

示例

>>> import mars.tensor as mt
>>> arr = mt.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
>>> arr.execute()
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])
>>> mt.delete(arr, 1, 0).execute()
array([[ 1,  2,  3,  4],
       [ 9, 10, 11, 12]])
>>> mt.delete(arr, np.s_[::2], 1).execute()
array([[ 2,  4],
       [ 6,  8],
       [10, 12]])
>>> mt.delete(arr, [1,3,5], None).execute()
array([ 1,  3,  5,  7,  8,  9, 10, 11, 12])