numpy array 将列表中负数换为nan

可以使用 numpy 的 where 函数将列表中的负数替换为 NaN，示例如下：

import numpy as np

lst = [1, -2, 3, -4, 5]
arr = np.array(lst)
arr = np.where(arr < 0, np.nan, arr)
print(arr)

输出结果为：

[ 1. nan  3. nan  5.]

其中，`np.where(arr < 0, np.nan, arr)` 表示将数组 arr 中小于 0 的元素替换为 NaN，大于等于 0 的元素不变。

相关问题

numpy.sqrt()

numpy.sqrt() 是一个numpy库中的函数，它用于计算数组中每个元素的平方根。它返回一个数组，该数组与输入数组具有相同的形状。请注意，输入数组中的负数元素将返回NaN。

使用示例：

import numpy as np

arr = np.array([4, 9, 16, 25])
sqrt_arr = np.sqrt(arr)

print(sqrt_arr) # [2. 3. 4. 5.]