mars.tensor.nanargmin#

mars.tensor.nanargmin(a, axis=None, out=None, combine_size=None)[来源]#

返回指定轴上最小值的索引，忽略NaN。对于全为NaN的切片会引发ValueError。警告：如果切片仅包含NaN和Infs，则结果不可信。

Parameters

  • a (array_like) – 输入数据。

  • axis (int, 可选) – 操作的轴。默认使用扁平化的输入。

  • combine_size (int, 可选) – 拼接的块数。

Returns

index_array – 一个索引的张量或单个索引值。

Return type

张量

另请参阅

argmin, nanargmax

示例

>>> import mars.tensor as mt

>>> a = mt.array([[mt.nan, 4], [2, 3]])
>>> mt.argmin(a).execute()
0
>>> mt.nanargmin(a).execute()
2
>>> mt.nanargmin(a, axis=0).execute()
array([1, 1])
>>> mt.nanargmin(a, axis=1).execute()
array([1, 0])