mindspore.numpy.nanmean

查看源文件
mindspore.numpy.nanmean(a, axis=None, dtype=None, keepdims=False)[源代码]

沿指定轴计算算术平均值，忽略NaN。

返回数组元素的平均值。默认情况下，平均值在展平的数组上计算，否则在指定轴上计算。对于整数输入，中间值和返回值使用float32类型。

说明

不支持NumPy参数 out 。

参数：

  • a (Union[int, float, list, tuple, Tensor]) - 包含要计算均值的数的数组。如果 a 不是数组，将尝试进行转换。

  • axis (Union[int, tuple of int, None], 可选) - 计算均值所沿的单个或多个轴。若取默认值，计算展平数组的均值。默认值: None 。

  • dtype (mindspore.dtype, 可选) - 默认值: None 。 覆盖输出Tensor的dtype。

  • keepdims (boolean, 可选) - 默认值: False 。如果设置为 True ，减少的轴在结果中保留为大小为1的维度。 若使用此选项，结果会广播到和 a 同一个维度数。

返回：

Tensor。

异常：

  • ValueError - 如果 axis 超出[-a.ndim, a.ndim)范围，或者 axis 包含重复项。

支持平台：

GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.array([[1, np.nan], [3, 4]])
>>> output = np.nanmean(a)
>>> print(output)
2.6666667
>>> output = np.nanmean(a, axis=0)
>>> print(output)
[2. 4.]
>>> output = np.nanmean(a, axis=1)
>>> print(output)
[1.  3.5]