Numpy Create Structured Array In Python Definition Function Math