How To Take An Array In Python