[Kwant] optimal way to compute LDOS for large sample