degree.py

from matplotlib import pyplot as plt
import math
import sys

plt.switch_backend('agg')

graph_name = sys.argv[1]       
# degree
with open('./pyplot/degree.txt', mode='rt') as f:
    while True:
        text=f.readline()
        if not text:
            break
        a=text.split()
        for i in range(len(a)):
            a[i]=int(a[i])

        b=[]
        for i in range(a[-1]+1):
            b.append(0)
    
        degree = list(range(0, a[-1]+1))
        for i in range(len(a)):
            b[a[i]]=b[a[i]]+1
    
        plt.figure(3)
        plt.scatter(degree, b)
        plt.xlabel('degree')
        plt.ylabel('# of vertex')
        plt.title('degree')
        fig = plt.gcf()
        fpath="./pyplot/"+str(graph_name)+"_degree.png"
        fig.savefig(fpath, dpi=300)

        #log scale
        plt.figure(4)
        plt.xscale('log')
        plt.yscale('log')
        xmax = max(degree) * 2
        ymax = max(b) * 2
        plt.xlim(0.8, xmax)
        plt.ylim(0.8, ymax)
        plt.scatter(degree, b)
        plt.xlabel('degree')
        plt.ylabel('# of vertex')
        plt.title('degree_log')
        fig = plt.gcf()
        fpath = "./pyplot/"+str(graph_name)+"_degree_log.png"
        fig.savefig(fpath, dpi=300)