import math import numpy as np def evaluate(inputs): outputs = np.empty([inputs.shape[0]]) for i, var in enumerate(inputs): outputs[i] = (var[0]-((2*var[1]*9.81)/(1.5*var[2]))) return outputs