numpy常用函数3

numpy常用函数3

标量方法矢量化

def 标量函数(标量参数1, 标量参数2, …):
…
return 标量返回值1, 标量返回值2, …
矢量参数1
矢量参数2
…
numpy.vectorize(标量函数)->矢量函数
也可以使用装饰器的方法：即在函数的方法前方加上@numpy.vectorize

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import numpy as np
@np.vectorize
def foo(x, y):
    return x + y, x - y, x * y
x, y = 1, 4
print(foo(x, y))
X, Y = np.array([1, 2, 3]), np.array([4, 5, 6])
print(foo(X, Y))