python - 基于字符串应用 numpy 函数

Question

是否可以应用基于字符串的 numpy 函数？如果我给 'max' 调用 np.max。

values = np.array([[1,2,-1],[2,3,6], [0,-1,4]]) 
aggregator = 'max'
print np.max(values, axis=0)
>>> [2 3 6]

我希望是这样的:

some_cool_function(aggregator, values, axis=0)
>>> [2 3 6]

这将提供更好的可读性并缩短我的代码。而不是做多个if。

编辑:

我找到了 numpy.apply_along_axis但它需要一个函数，它不能是一个字符串。

Answer 1

我想你在找getattr :

>>> getattr(np, 'max')(values, axis=0)
array([2, 3, 6])