pythonFLAC/colorbar2.py at main · chingchenn/pythonFLAC · GitHub

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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sat Jun  4 09:39:22 2022

@author: ji-chingchen
"""

import copy
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import matplotlib.colors as mcolors


plt.rcParams["font.family"] = "Times New Roman"
path='/home/jiching/geoflac/'
#path = '/scratch2/jiching/22winter/'
#path = '/scratch2/jiching/03model/'
#path = 'F:/model/'
# path = 'D:/model/'
path = '/Volumes/SSD500/model/'
savepath='/home/jiching/geoflac/data/'
savepath='/Volumes/SSD500/data/'
figpath='/home/jiching/geoflac/figure/'
# figpath='/Users/ji-chingchen/OneDrive - 國立台灣大學/年會/2022/POSTER/'
# figpath='/Users/chingchen/OneDrive - 國立台灣大學/Thesis_figure/colorbar/'
figpath='/Users/chingchen/Desktop/figure/'


### viscosity
fig, axes = plt.subplots(1, 1, figsize=(15, 1))
fig.subplots_adjust(wspace=2.5)
labelsize=40

cmap1 = copy.copy(cm.jet) # viscosity
cmap1 = copy.copy(cm.RdBu_r) # Pressure
norm1 = mcolors.Normalize(vmin=20, vmax=24)
im1 = cm.ScalarMappable(norm=norm1, cmap=cmap1)
cbar1 = fig.colorbar(im1, cax=axes,ticks=np.linspace(20, 24, 8),orientation='horizontal')
cbar1.ax.tick_params(labelsize=labelsize)
# cbar1.set_label('# of contacts', rotation=270,fontsize = 20)
# fig.savefig(figpath+'viscosity_hor.pdf')

### temperature
fig2, axes2 = plt.subplots(1,1, figsize=(15, 1))
bins = [0,200,400,600,800,1000,1200]
nbin = len(bins) - 1
cmap4 = cm.get_cmap('RdBu_r', nbin)
norm4 = mcolors.BoundaryNorm(bins, nbin)
im4 = cm.ScalarMappable(norm=norm4, cmap=cmap4)
cbar4 = fig.colorbar(im4, cax=axes2,orientation='horizontal')
cbar4.ax.tick_params(labelsize=labelsize)
# fig2.savefig(figpath+'temperature_hor.pdf')

### pressure
fig3, axes = plt.subplots(1, 1, figsize=(15, 1))
fig3.subplots_adjust(wspace=2.5)
labelsize=40
cmap1 = copy.copy(cm.jet) # viscosity
cmap1 = copy.copy(cm.RdBu_r) # Pressure
norm1 = mcolors.Normalize(vmin=-200, vmax=200)
im1 = cm.ScalarMappable(norm=norm1, cmap=cmap1)
cbar1 = fig3.colorbar(im1, cax=axes,ticks=np.linspace(-200, 200, 5),orientation='horizontal')
cbar1.set_label(label='Pressure (MPa)', size=40)
cbar1.ax.tick_params(labelsize=labelsize)
fig3.savefig(figpath+'pressure_hor.pdf')