def plot_distances()

in poincare_maps.py [0:0]

• 28 lines of code
• 6 McCabe index (conditional complexity)

    def plot_distances(self, cell=None, pm_type='rot', ss=10, eps=4.0, file_name=None, title_name=None, idx_zoom=None, show=False, fs=8, ms=3):
        if cell is None:
            cell = self.iroot
            
        if pm_type == 'ori':
            coordinates = self.coordinates
        elif pm_type == 'rot':
            coordinates = self.coordinates_rotated

        fig = plt.figure(figsize=(5, 5))
        circle = plt.Circle((0, 0), radius=1, color='black', fc="None")    
        cm = plt.cm.get_cmap('rainbow')
        
        mycmap = np.minimum(list(self.distances[:, cell]), eps)
        
        plt.gca().add_patch(circle)
        plt.plot(0, 0, 'x', c=(0, 0, 0), ms=ms)
        if title_name:
            plt.title(title_name, fontsize=fs)
        plt.scatter(coordinates[:, 0], coordinates[:, 1], c=mycmap, s=ss, cmap=cm)
        plt.plot(coordinates[cell, 0], coordinates[cell, 1], 'd', c='red')

        plt.plot(0, 0, 'x', c=(1, 1, 1), ms=ms)    
        plt.axis('off')
        plt.axis('equal')        
        
        plt.axis('off')
        plt.axis('equal')
        cb = plt.colorbar()
        cb.ax.tick_params(labelsize=fs) 


        if file_name:
            plt.savefig(file_name + '.pdf', format='pdf')
        
        plt.show()
        plt.close(fig)