OnDraganov

# Original code from: https://gist.github.com/fransua/da703c3d2ba121903c0de5e976838b71
# Minor edits to make it run in Python 3,
# and added label_func argument to control the displayed leaf names.

from itertools import chain

from matplotlib import pyplot as plt
from matplotlib.collections import LineCollection

import numpy as np