📜  如何仅平均数组 numpy 中的正数 - Python (1)

📅  最后修改于: 2023-12-03 15:08:13.505000             🧑  作者: Mango

如何仅平均数组 numpy 中的正数 - Python

当我们处理包含负数和正数的数组时,有时我们需要仅考虑正数的平均值。在 numpy 中,我们可以使用以下方法来实现这个目标。

首先,我们需要导入 numpy:

import numpy as np

假设我们有一个包含正数和负数的数组:

arr = np.array([1, -2, 3, -4, 5])

我们可以使用以下代码来获取仅包含 positive values 的子数组:

positive_arr = arr[arr > 0]

这是一个布尔索引,它将返回一个仅包含正数的新的 numpy 数组。现在,我们可以使用这个子数组来计算平均值:

average_positive = np.mean(positive_arr)

完整的代码如下:

import numpy as np

arr = np.array([1, -2, 3, -4, 5])
positive_arr = arr[arr > 0]
average_positive = np.mean(positive_arr)

print("Average of positive values:", average_positive)

这将打印出:

Average of positive values: 3.0

现在我们已经了解了如何仅平均 numpy 数组中的正数。这种方法可以应用于任何包含正数和负数的 numpy 数组。