Take Array From User In Python