-
Notifications
You must be signed in to change notification settings - Fork 0
/
plot_functions.py
65 lines (56 loc) · 2.24 KB
/
plot_functions.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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
64
65
""" Functions for computing the Fast Fourier Transform
Partially developed by Qikai Wu from The O'Hern Group at Yale University <https://jamming.research.yale.edu/>
"""
__author__ = 'Atoosa Parsa'
__credits__ = ['Atoosa Parsa', 'Qikai Wu']
__license__ = 'MIT License'
__version__ = '0.0.2'
__maintainer__ = 'Atoosa Parsa'
__email__ = '[email protected]'
__status__ = "Dev"
import matplotlib.pyplot as plt
def Line_single(xdata, ydata, line_spec, xlabel, ylabel, mark_print, fn = '', xscale='linear', yscale='linear'):
fig, ax1 = plt.subplots()
fig.set_size_inches(3.5,3.5*3/4)
pos1 = ax1.get_position()
pos2 = [pos1.x0 + 0.12, pos1.y0 + 0.05, pos1.width-0.1, pos1.height]
ax1.set_position(pos2)
#ax1.tick_params(labelsize=10)
for axis in ['top','bottom','left','right']:
ax1.spines[axis].set_linewidth(2)
plt.ylabel(ylabel, fontsize=12)
plt.xlabel(xlabel, fontsize=12)
plt.xscale(xscale), plt.yscale(yscale)
plt.style.use('default')
ax1.plot(xdata, ydata, line_spec)
if mark_print == 1:
fig.savefig(fn, dpi = 300)
fig.show()
def Line_multi(xdata, ydata, line_spec, xlabel, ylabel, xscale='linear', yscale='linear'):
fig, ax1 = plt.subplots()
fig.set_size_inches(3.5,3.5*3/4)
ax1.tick_params(labelsize=10)
for axis in ['top','bottom','left','right']:
ax1.spines[axis].set_linewidth(2)
ax1.set_ylabel(ylabel, fontsize=12)
plt.xlabel(xlabel, fontsize=12)
plt.xscale(xscale), plt.yscale(yscale)
plt.style.use('default')
for ii in range(len(xdata)):
ax1.plot(xdata[ii], ydata[ii], line_spec[ii])
plt.show()
def Line_yy(xdata, ydata, line_spec, xlabel, ylabel, xscale='linear', yscale='linear'):
fig, ax1 = plt.subplots()
fig.set_size_inches(3.5,3.5*3/4)
ax1.tick_params(labelsize=10)
for axis in ['top','bottom','left','right']:
ax1.spines[axis].set_linewidth(2)
ax1.set_ylabel(ylabel[0], fontsize=12)
plt.xlabel(xlabel, fontsize=12)
plt.xscale(xscale), plt.yscale(yscale)
plt.style.use('default')
ax1.plot(xdata[0], ydata[0], line_spec[0])
ax2 = ax1.twinx()
ax2.set_ylabel(ylabel[1], fontsize=12)
ax2.plot(xdata[1], ydata[1], line_spec[1])
plt.show()