np.where是啥

`np.where`是NumPy库中的一个功能强大的函数，它在数组中根据条件选择元素并返回一个新的数组。这个函数的基本用法是：

new_array = np.where(condition, value_if_true, value_if_false)

- `condition`: 这是一个布尔类型的数组，如果为True，则对应位置的值会被`value_if_true`替换；如果为False，则会被`value_if_false`替换。
- `value_if_true` 和 `value_if_false`: 分别是要给满足条件和不满足条件的数组元素设置的新值。

举个例子，如果你有一个数组`a`，你可以用`np.where`找出所有大于0的元素的位置，并将它们替换为1，其他元素替换为0：

a = [1, -2, 3, 0]
result = np.where(a > 0, 1, 0)
print(result)  # 输出：[1 0 1 0]