SouthernOceanPolynya/PlotMultiModels.py at master · shunzi-work/SouthernOceanPolynya · 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
###############
#
# # Python script for plotting polynya
#
###############

from myfunctions import *
from PlotFunctions import *
from mycolors import PALETTE as color_palette


import gc

def main():
    # filter some warning messages
    import warnings
    warnings.filterwarnings("ignore")

    datapd = pd.read_csv('List_model.csv')
    p_ice = '../../SO_data/data_siconc_w_area/'
    p_count = '../../SO_data/data_polynya_count/'
    p_polynya = '../../SO_data/data_polynya_mean/'
    p_mld = '../../SO_data/data_mld/'
    p_mlotst = '../../SO_data/data_mlotst/'

    # print('Plot histogram ...')
    # plot_ice_hist(datapd, p_ice, p_count, 'PolynyaCounts.pdf')
    # gc.collect()

    # print('Plot polynya maps (different thresholds) ...')
    # ice_thresholds = np.arange(0, 100, step=1)
    # savename_pre = 'Figures/polynya_maps_'
    # for ice_threshold in ice_thresholds:
    #     sn = savename_pre + str(ice_threshold) + '.png'
    #     if os.path.exists(sn):
    #         continue
    #     plot_polynya_maps(datapd, p_ice, color_palette, ice_threshold, savename_pre = savename_pre)
    #     gc.collect()

    print('Plot polynya maps (from polynya file) ...')
    plot_polynya_maps_from_polynya_data(datapd, p_ice, p_polynya, color_palette)
    gc.collect()

    print('Plot convection maps ...')
    plot_convection_maps(datapd, p_mld, p_mlotst, color_palette)
    gc.collect()


if __name__ == "__main__":
    main()