您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

在Python / Numpy中一次分配相同的数组索引

在Python / Numpy中一次分配相同的数组索引

是的,可以做到,但是有点棘手:

# convert yourmulti-dim indices to flat indices
flat_idx = np.ravel_multi_index((Px, Py), dims=a.shape)
# extract the unique indices and their position
unique_idx, idx_idx = np.unique(flat_idx, return_inverse=True)
# Aggregate the repeated indices 
deltas = np.bincount(idx_idx, weights=x)
# Sum them to your array
a.flat[unique_idx] += deltas
python 2022/1/1 18:50:48 有352人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶