Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • simplicity
  • v0.55
  • v0.54
  • v0.53
  • v0.52
  • v0.51
  • v0.50
  • v0.49
  • v0.48
  • v0.47
  • v0.46
  • v0.45
  • v0.44
  • v0.43
  • v0.42
  • v0.41
  • v0.40
  • v0.39
  • v0.38
  • v0.37
  • v0.36
22 results
Blame Permalink
distribution_function.py 4.62 KiB
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130 






#!/usr/bin/env python3

import shapefile
import numpy as np
import matplotlib.pyplot as plt
import itertools

def plot_distr_funct(x, label = None, ax = None):
    """Sort and plot distribution function. x should be a Numpy array."""
    
    x.sort()
    nx = np.size(x)
    
    if nx != 0:
        if ax is None: ax = plt.gca()
        ax.plot(x, (1 + np.arange(nx)) / nx, label = label)
        print("minimum value:", x[0])
        print("maximum value:", x[-1])
        
    print("number of values:", nx, "\n")

def read():
    """Read the three dbf files and return speed, radius and amplitude of
    outermost contour, radius and amplitude of maximum speed contour,
    as Numpy arrays.

    Select valid speed values and valid outermost contour (with
    sufficient area).

    """
    
    with shapefile.Reader("extremum") as extremum, \
         shapefile.Reader("outermost_contour") as outerm_cont, \
         shapefile.Reader("max_speed_contour") as max_speed_cont:
        speed = []
        rad_outer = []
        rad_speed = []
        amp_outer = []
        amp_speed = []

        for rec_extr, rec_outer, rec_max in zip(extremum.iterRecords(),
                                                outerm_cont.iterRecords(),
                                                max_speed_cont.iterRecords()):
            if rec_extr.speed != 1e4: speed.append(rec_extr.speed)

            if rec_extr.valid == 1:
                rad_outer.append(rec_outer.r_eq_area)
                amp_outer.append(rec_extr.ssh - rec_outer.ssh)
                
            if rec_max.r_eq_area != - 100:
                rad_speed.append(rec_max.r_eq_area)
                amp_speed.append(rec_extr.ssh - rec_max.ssh)

    return {"speed": np.array(speed), "rad_outer": np.array(rad_outer),
            "rad_speed": np.array(rad_speed), "amp_outer": np.array(amp_outer),
            "amp_speed": np.array(amp_speed)}

def fig_hist(xlabel, x, label):
    """Complete figure with histogram. x and label can also be sequences."""
    
    plt.figure()
    plt.hist(x, bins = "auto", histtype = "step", label = label)
    plt.xlabel(xlabel)
    plt.ylabel("number of occurrences")
    if label is not None: plt.legend()

def fig_distr_funct(xlabel, x, label = None):
    """Complete figure with distribution function. x must be a sequence of
    Numpy arrays and label None or a sequence of strings.

    """
    
    plt.figure()
    print(xlabel + ":")

    label_None = label is None
    if label_None: label = itertools.repeat(None)
        
    for xi, li in zip(x, label):
        plot_distr_funct(xi, label = li)
        
    plt.xlabel(xlabel)
    plt.ylabel("distribution function")
    if not label_None: plt.legend()
    
def plot_all(dict_list, label = None):
    """dict_list: list of dictionaries. label: list of labels, one label
    for each dictionary.

    """
    
    fig_hist("speed, in m s-1", [d["speed"] for d in dict_list], label)
    fig_distr_funct("speed, when positive, in m s-1",
                    [d["speed"][d["speed"] >= 0] for d in dict_list], label)
    fig_distr_funct("speed, when negative, in m s-1",
                    [- d["speed"][d["speed"] < 0] for d in dict_list], label)
    
    fig_hist("equivalent radius of outermost contour, in km",
             [d["rad_outer"] for d in dict_list], label)
    fig_distr_funct("equivalent radius of outermost contour, in km",
                    [d["rad_outer"] for d in dict_list], label)

    fig_hist("equivalent radius of max-speed contour, in km",
             [d["rad_speed"] for d in dict_list], label)
    fig_distr_funct("equivalent radius of max-speed contour, in km",
                    [d["rad_speed"] for d in dict_list], label)

    fig_hist("amplitude of outermost contour, in m",
             [d["amp_outer"] for d in dict_list], label)
    fig_distr_funct("amplitude of outermost contour, anticyclones, in m",
                    [d["amp_outer"][d["amp_outer"] >= 0] for d in dict_list],
                    label)
    fig_distr_funct("amplitude of outermost contour, cyclones, in m",
                    [- d["amp_outer"][d["amp_outer"] < 0] for d in dict_list],
                    label)

    fig_hist("amplitude of max-speed contour, in m",
             [d["amp_speed"] for d in dict_list], label)
    fig_distr_funct("amplitude of max-speed contour, anticyclones, in m",
                    [d["amp_speed"][d["amp_speed"] >= 0] for d in dict_list],
                    label)
    fig_distr_funct("amplitude of max-speed contour, cyclones, in m",
                    [- d["amp_speed"][d["amp_speed"] < 0] for d in dict_list],
                    label)
    
if __name__ == "__main__":
    d = read()
    plot_all([d])
    plt.show()