Kudos. Usually these type of "in N lines of Python"-type things proceed to call a 3rd-party library like "sklearn" or "matplotlib" to do all the heavy lifting. Fun, educational article.