代码之家 › 专栏 › 技术社区 › user

使用索引列表对numpy数组的元素执行操作

numpy python

user · 技术社区 · 7 年前

我有numpy数组和两个python索引列表,它们的位置可以将数组元素增加一个。numpy是否有一些方法可以在不使用 for 循环?

我目前执行缓慢:

a = np.zeros([4,5])
xs = [1,1,1,3]
ys = [2,2,3,0]

for x,y in zip(xs,ys): # how to do it in numpy way (efficiently)?
    a[x,y] += 1

print(a)

输出:

[[0. 0. 0. 0. 0.]
 [0. 0. 2. 1. 0.]
 [0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0.]]

2 回复 | 直到 7 年前

Tomas Farias 7 年前

np.add.at 只需将两个索引作为单个2d数组/列表传递即可:

a = np.zeros([4,5])
xs = [1, 1, 1, 3]
ys = [2, 2, 3, 0]

np.add.at(a, [xs, ys], 1) # in-place
print(a)

array([[0., 0., 0., 0., 0.],
       [0., 0., 2., 1., 0.],
       [0., 0., 0., 0., 0.],
       [1., 0., 0., 0., 0.]])

lenik 7 年前

>>> a = np.zeros([4,5])
>>> xs = [1, 1, 1, 3]
>>> ys = [2, 2, 3, 0]
>>> a[[xs,ys]] += 1
>>> a
array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  1.,  1.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 1.,  0.,  0.,  0.,  0.]])

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

6 月前

unfolx · numpy数组不等式的执行时间

7 月前

mchaudh4 · 用numpy表示三对角矩阵

7 月前

Geremia · 2D NumPy数组+1D数组?

7 月前

LMC · Numpy数组布尔索引以获取包含元素

7 月前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

8 月前

alpelito7 · 如何只获取numpy数组中每个递增值的第一次出现?

8 月前

Amarth Gûl · 找到一组向量的最近收敛点

8 月前

Mr. W · numpy.divide是函数、类还是其他什么?

8 月前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

8 月前