@H_403_1@我们有一个基于numpy的算法,应该处理不同类型的数据.
@H_403_1@
def my_fancy_algo(a):
b = np.sum(a,axis=1)
# Do something b
return b
*** ValueError: 'axis' entry is out of bounds
最佳答案
您的示例数组实际上提供了与标量相同的问题：
@H_403_1@
>>> a = np.array([1.0,3.0])
>>> np.sum(a,axis=1)
Traceback (most recent call last):
File "np.atleast_2d：
@H_403_1@
>>> np.sum(np.atleast_2d(a),axis=1)
array([ 6.])
>>> np.sum(np.atleast_2d(6.0),axis=1)
array([ 6.])
>>> np.sum(a)
6.0
>>> np.sum(6.0)
6.0