Skip to content

Plotting Utils

Plotting Utils

plot_density_map(data, y_title, filename=None)

Plot a density map using fig and ax Parameters: data: A list of desired variable to be plotted y_title: Title of the y axis filename: Name of the output file name, containing the path

Source code in tgx/utils/plotting_utils.py 
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
def plot_density_map(data: list, 
                     y_title: str,
                     filename: str = None,):
    '''
    Plot a density map using fig and ax
    Parameters:
        data: A list of desired variable to be plotted
        y_title: Title of the y axis
        filename: Name of the output file name, containing the path
    '''
    max_value = max(max(inner) for inner in data if inner)
    c = np.zeros((max_value, len(data)))

    for i, row in enumerate(data):
        for value in row:
            c[value - 1][i] += 1

    # Plot
    fig = plt.figure(facecolor='w', figsize=(9, 6))
    ax = fig.add_subplot(111)

    norm = mcolors.Normalize(vmin=0, vmax=1)
    cax = ax.imshow(c, cmap='viridis', interpolation='nearest', norm=norm)
    cbar = fig.colorbar(cax)
    cbar.set_label('Frequency')

    ax.set_title("Heatmap of Node Degrees Over Time")
    ax.set_xlabel('Time', fontsize=20)
    ax.set_ylabel(y_title, fontsize=20)
    ax.tick_params(labelsize=20)
    ax.xaxis.set_major_locator(MaxNLocator(integer=True))

    # Adjust the aspect ratio of the plot
    ax.set_aspect('auto')

    if filename is not None:
        plt.savefig(f'{filename}')
    else:
        plt.show()

plot_for_snapshots(data, y_title, filename=None, show_ave=True)

Plot a variable for different timestamps Parameters: data: A list of desired variable to be plotted y_title: Title of the y axis filename: Name of the output file name, containing the path show_ave: Whether to plot a line showing the average of the variable over all timestamps

Source code in tgx/utils/plotting_utils.py 
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
def plot_for_snapshots(data: list,  
                       y_title: str, 
                       filename: str = None, 
                       show_ave: bool=True, ):
    '''
    Plot a variable for different timestamps
    Parameters:
        data: A list of desired variable to be plotted
        y_title: Title of the y axis
        filename: Name of the output file name, containing the path
        show_ave: Whether to plot a line showing the average of the variable over all timestamps
    '''
    ts = list(range(0, len(data)))
    # plt.rcParams["font.family"] = "Times New Roman"
    fig = plt.figure(facecolor='w', figsize=(9,6))
    ax = fig.add_subplot(111)
    ax.plot(ts, data, color='black', lw=3)

    ax.set_xlabel('Time', fontsize=20)
    ax.set_ylabel(y_title, fontsize=20)
    ax.tick_params(labelsize=20)
    ax.set_xlim(0, len(ts)-1)
    if show_ave:
        ave_deg = [np.average(data) for i in range(len(ts))]
        ax.plot(ts, ave_deg, color='#ca0020', linestyle='dashed', lw=3)
    if filename is not None:
        plt.savefig(f'{filename}')
    else:
        plt.show()

plot_nodes_edges_per_ts(edges, nodes, ts, filename=None, ylabel_1='Edges per Timestamp', ylabel_2='Nodes per Timestamp')

Plot nodes and edges per timestamp in one figure Parameters: edges: A list containing number of edges per timestamp nodes: A list containing number of nodes per timestamp ts: list of timestamps filename: Name of the output file name, containing the path ylabel_1: Label for the edges per timestamp line ylabel_2: Label for the nodes per timestamp line

Source code in tgx/utils/plotting_utils.py 
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
def plot_nodes_edges_per_ts(edges: list,
                            nodes: list, 
                            ts: list,
                            filename: str = None,
                            ylabel_1: str = 'Edges per Timestamp',
                            ylabel_2: str = 'Nodes per Timestamp'):
    """
    Plot nodes and edges per timestamp in one figure
    Parameters:
        edges: A list containing number of edges per timestamp
        nodes: A list containing number of nodes per timestamp
        ts: list of timestamps
        filename: Name of the output file name, containing the path
        ylabel_1: Label for the edges per timestamp line
        ylabel_2: Label for the nodes per timestamp line
    """
    fig = plt.figure(facecolor='w', figsize=(11, 6))
    ax1 = fig.add_subplot(111)
    ax2 = ax1.twinx()

    c1, = ax1.plot(ts, edges, color='black', lw=3, label=ylabel_1)
    c2, = ax2.plot(ts, nodes, color='gray', linestyle='dashed', lw=3, label=ylabel_2)
    curves = [c1, c2]
    ax1.legend(curves, [curve.get_label() for curve in curves], fontsize = 18)
    ax1.set_xlabel('Time', fontsize=20)
    ax1.set_ylabel(ylabel_1, fontsize=20)
    ax2.set_ylabel(ylabel_2, fontsize=20)
    ax1.tick_params(labelsize=20)
    ax2.tick_params(labelsize=20)
    ax1.set_ylim(0)
    ax2.set_ylim(0)
    ax1.set_xlim(0, len(ts)-1)
    if filename is not None:
        plt.savefig(f'{filename}')
    else:
        plt.show()