Return An Array In Python