forked from weecology/sad-comparison
/
example-plot.py
78 lines (64 loc) · 2.99 KB
/
example-plot.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
66
67
68
69
70
71
72
73
74
75
76
77
78
from __future__ import division
import numpy as np
import macroeco_distributions as md
from scipy.stats import zipf
import matplotlib
import matplotlib.pyplot as plt
#%matplotlib inline
# BBS 61036
# fia site 14057000012
# Gentry site 95
ab_array = np.array([[1L, 15L, 3L, 1L, 1L, 1L, 1L, 8L, 2L, 3L, 1L, 1L, 5L, 3L, 1L,
7L, 6L, 2L, 5L, 2L, 1L, 6L, 28L, 12L, 30L, 2L, 8L, 3L, 7L, 9L,
1L, 2L, 1L, 13L, 1L, 4L, 4L, 7L, 26L, 10L, 1L, 28L, 5L, 4L, 14L,
11L, 3L, 4L, 2L, 4L, 1L, 20L, 1L, 4L, 17L, 1L, 2L, 5L, 3L, 2L,
14L, 1L, 2L, 3L, 3L, 4L, 5L, 18L, 30L, 5L, 5L, 6L, 7L, 3L, 2L,
4L],
[2L, 1L, 1L, 2L, 1L, 2L, 1L, 5L, 1L, 1L, 1L, 4L, 7L, 2L, 1L,
2L, 1L, 3L, 5L, 1L, 1L],
[4L, 22L, 7L, 12L, 2L, 6L, 2L, 15L, 5L, 6L, 1L, 3L, 1L, 2L,
3L, 6L, 1L, 3L, 2L, 3L, 9L, 5L, 8L, 2L, 5L, 1L, 5L, 7L, 4L, 1L,
4L, 3L, 4L, 1L, 2L, 3L, 11L, 5L, 7L, 3L, 1L, 2L, 1L, 1L, 1L,
4L, 1L, 1L, 9L, 3L, 8L, 3L, 2L, 10L, 4L, 14L, 4L, 2L, 3L, 1L,
1L, 8L]])
# Color scheme from Dave Harris in
# https://github.com/weecology/sad-comparison/issues/217
colors = ["#990F0F", "#99700F", "#1F990F", "#710F99"]
plot_labels = ["A", "B", "C"]
fig_example = plt.figure(figsize = (12, 4))
for i in range(3):
ax = plt.subplot(1, 3, i + 1)
ab = ab_array[i]
x_values = np.array(range(max(ab) + 2)[1:])
logser_p = md.logser_solver(ab)
logser_values = md.trunc_logser.pmf(x_values, logser_p, upper_bound=float("inf"))
lsll = md.logser_ll(ab, logser_p)
nb_n, nb_p = md.nbinom_lower_trunc_solver(ab)
nb_values = md.nbinom_lower_trunc.pmf(x_values, nb_n, nb_p)
nbll = md.nbinom_lower_trunc_ll(ab, nb_n, nb_p)
pln_mu, pln_sigma = md.pln_solver(ab)
pln_values = md.pln.pmf(x_values, pln_mu, pln_sigma, lower_trunc=True)
plnll = md.pln_ll(ab, pln_mu, pln_sigma)
zipf_par = md.zipf_solver(ab)
zipf_values = zipf.pmf(x_values, zipf_par)
zll = md.zipf_ll(ab, zipf_par)
ab_y = np.zeros(len(x_values) + 1)
for j in range(len(ab)):
ab_y[ab[j]] = ab_y[ab[j]] + 1/len(ab)
ax.set_xlim([0,min(50, max(x_values))])
plt.ylabel('frequency')
plt.xlabel('abundance')
plt.title(plot_labels[i])
# Width originally set at 12 when width was 50.
# This should be the same proportional width
width = 3 / min(50, max(x_values)) * 50
ax.vlines(x = x_values, ymin = [0], ymax = .0025, color = "#AAAAAA")
ax.vlines(x = x_values, ymin=[0], ymax=ab_y[1:], linewidth = width, color = "#AAAAAA")
ax.plot(x_values, logser_values, color = colors[0], label = "log-series: " + str(round(lsll)), linewidth = 3)
ax.plot(x_values, nb_values, color = colors[1], label = "negative binomial: " + str(round(nbll)), linewidth = 3)
ax.plot(x_values, pln_values, color = colors[2], label = "Poisson lognormal: " + str(round(plnll)), linewidth = 3)
ax.plot(x_values, zipf_values, color = colors[3], label = "Zipf: " + str(round(zll)), linewidth = 3)
plt.legend(loc='upper right', prop = {'size': 10})
plt.tight_layout()
plt.savefig('Fig1.pdf', dpi = 800)
plt.savefig('Fig1.png', dpi = 800)