import topotoolbox
import matplotlib.pyplot as plt
dem = topotoolbox.load_dem('bigtujunga')
fd = topotoolbox.FlowObject(dem)
D = fd.drainagebasins()
_= D.shufflelabel().plot(cmap="Pastel1",interpolation="nearest")
plt.show()