计算行均值时忽略0值

以下是一个示例代码，用于计算行均值时忽略0值：

import numpy as np

# 创建一个包含0值的二维数组
arr = np.array([[1, 2, 3, 0],
                [4, 5, 6, 0],
                [7, 8, 9, 0]])

# 计算行均值时忽略0值
row_mean = np.ma.masked_equal(arr, 0).mean(axis=1)

print(row_mean)

输出结果：

[2. 5. 8.]

在上面的代码中，我们使用了NumPy库的ma模块。np.ma.masked_equal()函数创建了一个掩码数组，将原始数组中的0值视为缺失值，并将其屏蔽。然后，我们可以使用mean()函数计算行均值，忽略了被掩码屏蔽的0值。最后，我们打印了计算得到的行均值。