温馨提示×

numpy怎么替换数组中的某一个数值

小亿
264
2023-11-29 20:05:52
栏目: 编程语言

要替换NumPy数组中的特定值,可以使用NumPy的where函数。

以下是一个示例代码,演示如何将数组中的某个特定值替换为新值:

import numpy as np

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

# 将数组中的值为2的元素替换为10
new_arr = np.where(arr == 2, 10, arr)

print(new_arr)

输出:

[ 1 10  3  4  5]

在上述示例中,使用了np.where函数来查找数组arr中的值为2的元素,并将其替换为10。替换后的新数组存储在new_arr中。

你可以根据需要自定义替换的条件和替换的新值。

0