是否存在一种矢量化(或更好的)方式,可以基于除此方式以外的其他方式为numpy数组的某些数据点设置值?
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
pos = np.array([[1, 2], [2, 0]])
for p in pos:
i,j = p
data[i,j] = 20
print(data)
在更高版本的Python中,您可以通过解压缩另一个可迭代对象来在理解范围内创建列表。然后,我们传递该列表以进行切片分配。
我们访问(切片)的方式是通过整数数组索引完成的
data[[*pos]] = 20
data
array([[ 1, 2, 3],
[ 4, 5, 20],
[20, 8, 9]])
对于其他版本的Python,请尝试:
data[pos.tolist()] = 20
data
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句