python - ndarray 如何处理一个简单的函数，比如 x**2？

Question

ndarray 如何处理一个简单的函数，比如 x**2？我收到此错误:

arr = array([3,4,5])
f = lambda x: x**2
print f(arr)     # works, but how?
print f(3)       # works
print f([3,4,5]) # will not work
#TypeError: unsupported operand type(s) for ** or pow(): 'list' and 'int'
print f((3,4,5)) # will not work
#TypeError: unsupported operand type(s) for ** or pow(): 'tuple' and 'int'

>>>f(arr)
#[ 9 16 25]
>>>f(3)
#9

终于……

class Info(object):
    def __init__(self,x):
        self.a = x<
    def __pow__(self,x):
        for i in xrange(len(self.a)):
            self.a[i]**=x
        return self

a = Info([3,4])
f = lambda x:x**2
a = f(a)
print a.a
[9, 16]

Answer 1

因为 ndarray defines the behaviour对于 ** 运算符，而列表元组则不然(它们可能包含任何对象，而 ndarrays 通常用于数字)。