CLN/TST: remove unused imports and correct the mpl.use function usage
DOC: groupby vis docs
DOC: clean up the example a bit