numpy.ma.where()函数 - Python

numpy.ma.where()函数是numpy中的一个函数，用于根据条件返回输入数组中的元素。

语法

numpy.ma.where(condition[, x, y])

参数

• condition：一个布尔数组，与输入数组大小相同。
• x：可选参数，输出数组中的替换值。它必须与 y 参数一起使用，或者不使用。
• y：可选参数，输出数组中的替换值。它必须与 x 参数一起使用，或者不使用。

返回值

numpy.ma.where()函数返回一个根据condition中的真值返回相应值的数组。

示例

import numpy.ma as ma

a = ma.array([1, 2, 3, 4])
b = ma.array([5, 6, 7, 8])

# 返回一个由 x、y 元素组成的数组，如果 condition 中的值为真，则取 x，否则取 y
result = ma.where(a > 2, a, b)

print(result)

输出：

[5 6 3 4]

说明：在这个例子中，condition是a>2，所以如果a中的元素大于2，则返回a中该位置的元素，否则返回b中该位置的元素。

总结

numpy.ma.where()函数是使用条件来选择数组元素的一个方便的工具。使用它可以使得对数组的处理更快捷、高效。