45 lines
1002 B
Python
45 lines
1002 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
import matplotlib.lines as lines
|
||
|
|
|
||
|
|
def set_size(width, fraction=1):
|
||
|
|
""" Set aesthetic figure dimensions to avoid scaling in latex.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
width: float
|
||
|
|
Width in pts (\\the\\textwidth)
|
||
|
|
fraction: float
|
||
|
|
Fraction of the width which you wish the figure to occupy
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
fig_dim: tuple
|
||
|
|
Dimensions of figure in inches
|
||
|
|
"""
|
||
|
|
# Width of figure
|
||
|
|
fig_width_pt = width * fraction
|
||
|
|
|
||
|
|
# Convert from pt to inches
|
||
|
|
inches_per_pt = 1 / 72.27
|
||
|
|
|
||
|
|
# Golden ratio to set aesthetic figure height
|
||
|
|
golden_ratio = (5 ** 0.5 - 1) / 2
|
||
|
|
|
||
|
|
# Figure width in inches
|
||
|
|
fig_width_in = fig_width_pt * inches_per_pt
|
||
|
|
# Figure height in inches
|
||
|
|
fig_height_in = fig_width_in * golden_ratio
|
||
|
|
|
||
|
|
return fig_width_in, fig_height_in
|
||
|
|
|
||
|
|
|
||
|
|
def config_plt():
|
||
|
|
plt.rc('font', size=11, family='Latin Modern Roman')
|
||
|
|
plt.rc('text', usetex=True)
|
||
|
|
plt.rc('xtick', labelsize=11)
|
||
|
|
plt.rc('ytick', labelsize=11)
|
||
|
|
plt.rc('axes', labelsize=11)
|