温馨提示×

numpy对矩阵元素怎么赋值

小亿
152
2023-08-19 02:35:15
栏目: 编程语言

使用NumPy可以通过索引来赋值矩阵元素。具体的赋值操作取决于你想要赋值的矩阵的维度。

对于一维数组,可以使用索引来赋值元素。例如:

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
arr[0] = 10
print(arr)

输出为:[10 2 3 4 5]

对于二维数组,可以使用索引来赋值元素。例如:

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
arr[0, 1] = 10
print(arr)

输出为:

[[ 1 10  3]
[ 4  5  6]]

你还可以使用切片来赋值多个元素。例如:

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
arr[1:4] = 10
print(arr)

输出为:[ 1 10 10 10 5]

你也可以使用布尔索引来赋值元素。例如:

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
mask = arr > 2
arr[mask] = 10
print(arr)

输出为:[ 1 2 10 10 10]

0