利用python进行数据分析-numpy练习（一）

import numpy as np

# 测试1
# def func(a,b):
#     def func1(x):
#         d = a+b+x
#         return d
#     return np.frompyfunc(func1, 1, 1)

#  frompyfunc() 第一个参数表示函数，第二个表示输入参数个数，第三个表示输出参数个数
# 测试2
def func(r):
    def func1(a, b):
        if a < b:
            t = a + b + r
        if a >= b:
            t = a - b + r
        return t
    return np.frompyfunc(func1, 2, 1)

arr1 = np.array([1, 2, 5, 6])
brr1 = np.array([3, 4, 1, 2])
y = func(1)(arr1,brr1)
print(y)