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

依次在numpy数组中添加值,而无需循环

依次在numpy数组中添加值,而无需循环

用途numpy.add.at

>>> import numpy as np
>>> A = np.array([1,2,3])
>>> B = np.array([10,20,30])
>>> I = np.array([0,1,1])
>>> 
>>> np.add.at(A, I, B)
>>> A
array([11, 52,  3])

或者,np.bincount

>>> A = np.array([1,2,3])
>>> B = np.array([10,20,30])
>>> I = np.array([0,1,1])
>>> 
>>> A += np.bincount(I, B, minlength=A.size).astype(int)
>>> A
array([11, 52,  3])

哪个更快?

依靠。在这个具体示例中,add.at速度似乎稍快一些,大概是因为我们需要在bincount解决方案中转换类型。

如果OTOHABfloatD类,然后bincount会更快。

其他 2022/1/1 18:46:23 有521人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

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

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

请先登录

推荐问题


联系我
置顶