Visualization of clusterization in Python