Numpy Create New Array From Existing Array In Math