Python Intro - Numpy function vectorize


#Numpy vectorize  generalized function class



#!/usr/local/bin/python3

def myfunc(a, b):
   if(a > b): return a;
   else:      return b;

try:
    import numpy as np
except ImportError:
    print("numpy is not installed")

vfunc = np.vectorize(myfunc);

result=vfunc([1, 6, 3, 9, 0], 5);

print(result);




[5 6 5 9 5]



你可能感兴趣的:(Python Intro - Numpy function vectorize)