numpy怎么替換數(shù)組中的某一個(gè)數(shù)值

小億
249
2023-11-29 20:05:52

要替換NumPy數(shù)組中的特定值，可以使用NumPy的where函數(shù)。

以下是一個(gè)示例代碼，演示如何將數(shù)組中的某個(gè)特定值替換為新值：

import numpy as np

# 創(chuàng)建一個(gè)示例數(shù)組
arr = np.array([1, 2, 3, 4, 5])

# 將數(shù)組中的值為2的元素替換為10
new_arr = np.where(arr == 2, 10, arr)

print(new_arr)

輸出：

[ 1 10  3  4  5]

在上述示例中，使用了np.where函數(shù)來(lái)查找數(shù)組arr中的值為2的元素，并將其替換為10。替換后的新數(shù)組存儲(chǔ)在new_arr中。

你可以根據(jù)需要自定義替換的條件和替換的新值。

0 踩