Python | numpy.putmask() 方法

简介

Python的NumPy模块中的numpy.putmask()方法用于通过boolean掩码来替换一个数组中的元素。

语法：numpy.putmask(arr, mask, values)

参数

• arr：输入的数组
• mask：Boolean掩码数组，制定哪些元素需要替换
• values：用于替换的数值或数组

返回值

• None

示例

import numpy as np

# 创建一个数组
a = np.array([1, 2, 3, 4])

# 创建一个Boolean掩码数组
mask = np.array([True, False, True, False])

# 将所有掩码为True的元素替换为-1
np.putmask(a, mask, -1)

print(a)

输出结果：

[-1 2 -1 4]

在上面的示例代码中，我们首先使用NumPy创建了一个数组a，然后创建一个Boolean掩码数组mask，并将掩码为True的元素替换为-1，并在控制台输出最终结果。

结论

numpy.putmask()方法是用于数组中元素替换的强大方法。我们可以使用Boolean掩码数组来指定要替换的元素，并指定要替换的值或数组。