为旧的NumPy版本重新实现numpy.add.at

时间:2017-07-19 18:12:45

标签: python numpy

我正在使用一个调用np.add.at的库,但我的numpy版本(1.7.1)太旧了,它不包含此功能。

不幸的是,我的代理人有不同的语义:

def np_add_at(a, ix, b):
   a[ix] += b

a = np.array([1])
np_add_at(a, [0,0], 1)
# a is now array([2])
b = np.array([1])
np.add.at(b, [0,0], 1)
# b is now array([3])

有没有办法告诉我这个问题是否出现并抛出异常? IE浏览器。我想检测ix参数是否有任何重复,至少会抛出错误。

更糟糕的是,调用np.add.at的方式,ix通常采用(None, slice(None, None, None))等值。

0 个答案:

没有答案