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
|
# This program is public domain
# Author: Paul Kienzle
"""
Table plotter
"""
__all__ = ['table_plot']
def table_plot(data, form="line", label=None, title=None):
"""
Plot periodic table data using element symbol vs. value.
:Parameters:
*data* : { Element: float }
Data values to plot
*form* = "line" : "line|grid"
Table layout to use
:Returns: None
"""
import matplotlib.pyplot as plt
if form == "line":
bbox = dict(boxstyle="round", lw=1, ec=(0, 0, 0), fc=(0.85, 0.8, 0.8))
for el, value in data.items():
if value is not None:
plt.text(el.number, value, el.symbol,
bbox=bbox, va='center', ha='center')
plt.xlim(0, 100)
plt.xlabel('Element number')
values = [v for v in data.values()]
minv, maxv = min(values), max(values)
margin = (maxv - minv)*0.05
plt.ylim(minv-margin, maxv+margin)
if label is not None:
plt.ylabel(label)
if title is not None:
plt.title(title)
|
