How To Push Function Into Array In Python