-
Notifications
You must be signed in to change notification settings - Fork 906
/
Copy pathextents.py
72 lines (59 loc) · 2.78 KB
/
extents.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
# -----------------------------------------------------------------------------
# Matplotlib cheat sheet
# Released under the BSD License
# -----------------------------------------------------------------------------
import sys
import pathlib
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
ROOT_DIR = pathlib.Path(__file__).parent.parent
sys.path.append(str(ROOT_DIR / "fonts"))
import custom_fonts # noqa
mpl.style.use([
ROOT_DIR / 'styles/base.mplstyle',
])
mpl.rc('figure.constrained_layout', wspace=0.05)
Z = np.arange(5*5).reshape(5, 5)
(fig, axs) = plt.subplots(figsize=(8, 5), nrows=2, ncols=2)
ax = axs[0, 0]
ax.imshow(Z, extent=[0, 10, 0, 5], interpolation="nearest", origin="upper")
ax.set_xlim(-1, 11), ax.set_xticks([])
ax.set_ylim(-1, 6), ax.set_yticks([0, 5])
ax.text(1, 4.5, "(0,0)", ha="center", va="center", color="white", size="large")
ax.text(9, 0.5, "(4,4)", ha="center", va="center", color="black", size="large")
ax.text(5.0, 5.5, 'origin="upper"',
ha="center", va="center", color="black", size="large")
ax.text(5.0, -0.5, "extent=[0,10,0,5]",
ha="center", va="center", color="black", size="large")
ax = axs[1, 0]
ax.imshow(Z, extent=[0, 10, 0, 5], interpolation="nearest", origin="lower")
ax.set_xlim(-1, 11), ax.set_xticks([0, 10])
ax.set_ylim(-1, 6), ax.set_yticks([0, 5])
ax.text(1, 0.5, "(0,0)", ha="center", va="center", color="white", size="large")
ax.text(9, 4.5, "(4,4)", ha="center", va="center", color="black", size="large")
ax.text(5.0, 5.5, 'origin="lower"',
ha="center", va="center", color="black", size="large")
ax.text(5.0, -0.5, "extent=[0,10,0,5]",
ha="center", va="center", color="black", size="large")
ax = axs[1, 1]
ax.imshow(Z, extent=[10, 0, 0, 5], interpolation="nearest", origin="lower")
ax.set_xlim(-1, 11), ax.set_xticks([0, 10])
ax.set_ylim(-1, 6), ax.set_yticks([])
ax.text(9, 0.5, "(0,0)", ha="center", va="center", color="white", size="large")
ax.text(1, 4.5, "(4,4)", ha="center", va="center", color="black", size="large")
ax.text(5.0, 5.5, 'origin="lower"',
ha="center", va="center", color="black", size="large")
ax.text(5.0, -0.5, "extent=[10,0,0,5]",
ha="center", va="center", color="black", size="large")
ax = axs[0, 1]
ax.imshow(Z, extent=[10, 0, 0, 5], interpolation="nearest", origin="upper")
ax.set_xlim(-1, 11), ax.set_xticks([])
ax.set_ylim(-1, 6), ax.set_yticks([])
ax.text(9, 4.5, "(0,0)", ha="center", va="center", color="white", size="large")
ax.text(1, 0.5, "(4,4)", ha="center", va="center", color="black", size="large")
ax.text(5.0, 5.5, 'origin="upper"',
ha="center", va="center", color="black", size="large")
ax.text(5.0, -0.5, "extent=[10,0,0,5]",
ha="center", va="center", color="black", size="large")
fig.savefig(ROOT_DIR / "figures/extents.pdf", dpi=600)