def plot_markers()

in poincare_maps.py [0:0]

• 36 lines of code
• 11 McCabe index (conditional complexity)

    def plot_markers(self, data, markesnames, pm_type='rot', file_name=None, fs=8, sc=3):
        if pm_type == 'ori':
            poincare_coord = self.coordinates
        elif pm_type == 'rot':
            poincare_coord = self.coordinates_rotated

        n_plt = np.size(data, 1)    
        
        # n2 = int(np.sqrt(n_plt))
        n2 = 3
        n1 = n_plt // n2
        
        if n1*n2 < n_plt:
            n1 += 1

        if n1 == 1:
            n1 = 2
        
        if n2 == 1:
            n2 = 2 
        
        f, axs = plt.subplots(n1, n2, sharey=False, figsize=(n2*sc, n1*sc))

        cm = plt.cm.get_cmap('jet')

        l=0
        for i in range(n1):
            for j in range(n2):            
                axs[i, j].axis('off')
                axs[i, j].axis('equal')            
                if l < n_plt:
                    circle = plt.Circle((0, 0), radius=1, color='black', fc="none")
                    axs[i, j].add_patch(circle)
                    axs[i, j].plot(0, 0, 'x', c=(0, 0, 0), ms=3)
                    axs[i, j].axis('equal')                 
                    sc = axs[i, j].scatter(poincare_coord[:, 0], poincare_coord[:, 1], c=data[:, l], s=5, cmap=cm)    
                    axs[i, j].set_title(markesnames[l], fontsize=fs)
                    plt.colorbar(sc,ax=axs[i,j])

                    if l == n_plt:
                        axs[i, j].legend(np.unique(labels), loc='center left', bbox_to_anchor=(1, 0.5), fontsize=fs)
                l+=1

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


        plt.show()
        plt.close(f)