summaryrefslogtreecommitdiff
path: root/stats/gini_index.py
blob: 8821429fa356563d8602aae94885fcfc62c5d06b (plain)
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
import numpy as np


def gini(x):
    mean = np.mean(x)
    mad = np.mean(np.abs(x - mean))
    rmad = mad/mean
    return rmad/2


if __name__ == '__main__':

    x = np.asarray([2, 2, 4, 4])
    print(gini(x))
    x = np.asarray([1, 1, 6, 4])
    print(gini(x))

    x = np.random.rand(500)
    print(gini(x))
    x = np.random.rand(500)+1
    print(gini(x))
    x = np.random.rand(500)+10
    print(gini(x))
    x = np.random.rand(500)+100
    print(gini(x))