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
|
# Copyright (c) 2017 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""General purpose utilities.
"""
import numpy as np
from os import walk
from os.path import join
from math import sqrt
def mean(items):
"""Calculate mean value from the items.
:param items: Mean value is calculated from these items.
:type items: list
:returns: MEan value.
:rtype: float
"""
return float(sum(items)) / len(items)
def stdev(items):
"""Calculate stdev from the items.
:param items: Stdev is calculated from these items.
:type items: list
:returns: Stdev.
:rtype: float
"""
avg = mean(items)
variance = [(x - avg) ** 2 for x in items]
stddev = sqrt(mean(variance))
return stddev
def relative_change(nr1, nr2):
"""Compute relative change of two values.
:param nr1: The first number.
:param nr2: The second number.
:type nr1: float
:type nr2: float
:returns: Relative change of nr1.
:rtype: float
"""
return float(((nr2 - nr1) / nr1) * 100)
def remove_outliers(input_data, outlier_const):
"""
:param input_data: Data from which the outliers will be removed.
:param outlier_const: Outlier constant.
:type input_data: list
:type outlier_const: float
:returns: The input list without outliers.
:rtype: list
"""
data = np.array(input_data)
upper_quartile = np.percentile(data, 75)
lower_quartile = np.percentile(data, 25)
iqr = (upper_quartile - lower_quartile) * outlier_const
quartile_set = (lower_quartile - iqr, upper_quartile + iqr)
result_lst = list()
for y in data.tolist():
if quartile_set[0] <= y <= quartile_set[1]:
result_lst.append(y)
return result_lst
def get_files(path, extension=None, full_path=True):
"""Generates the list of files to process.
:param path: Path to files.
:param extension: Extension of files to process. If it is the empty string,
all files will be processed.
:param full_path: If True, the files with full path are generated.
:type path: str
:type extension: str
:type full_path: bool
:returns: List of files to process.
:rtype: list
"""
file_list = list()
for root, _, files in walk(path):
for filename in files:
if extension:
if filename.endswith(extension):
if full_path:
file_list.append(join(root, filename))
else:
file_list.append(filename)
else:
file_list.append(join(root, filename))
return file_list
def get_rst_title_char(level):
"""Return character used for the given title level in rst files.
:param level: Level of the title.
:type: int
:returns: Character used for the given title level in rst files.
:rtype: str
"""
chars = ('=', '-', '`', "'", '.', '~', '*', '+', '^')
if level < len(chars):
return chars[level]
else:
return chars[-1]
|
