-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathmean.py
More file actions
38 lines (30 loc) · 865 Bytes
/
mean.py
File metadata and controls
38 lines (30 loc) · 865 Bytes
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
import numpy as np
def mean_fdt(x):
# Definir intervalos de classe com base nos valores 'start', 'end' e 'h'
breaks = np.arange(x['breaks']['start'], x['breaks']['end'] + x['breaks']['h'], x['breaks']['h'])
# Calcular pontos médios dos intervalos de classe
mids = 0.5 * (breaks[:-1] + breaks[1:])
# Frequências das classes
y = x['table'][:, 1]
# Calcular a média ponderada dos pontos médios
res = np.sum(y * mids) / np.sum(y)
# Retornar a média
return res
'''# Exemplo de entrada de teste para mean_fdt
x_test = {
'breaks': {
'start': 0,
'end': 40,
'h': 10
},
'table': np.array([
[1, 5],
[2, 10],
[3, 15],
[4, 10]
])
}
# Executar a função e imprimir o resultado
resultado = mean_fdt(x_test)
print("Média:", resultado)
'''