Flatten Array In Python