📅  最后修改于: 2023-12-03 15:08:13.505000             🧑  作者: Mango
当我们处理包含负数和正数的数组时,有时我们需要仅考虑正数的平均值。在 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 数组。