BigDataAnalytics/Cognitive_Diversity.py at master · icgranger9/BigDataAnalytics · 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
import pandas as pd
import math

# use diversity as defined in Eye Movement paper page 337 part B
def calculateDiversity(data, attrsA, attrsB):
    sum = 0.0
    for i in range(data.shape[0]):
        aSum = 0.0
        aMax = data.iloc[0, attrsA[0]]
        aMin = aMax
        aTempSum = 0.0

        bSum = 0.0
        bMax = data.iloc[0, attrsB[0]]
        bMin = bMax
        bTempSum = 0.0

        for n in attrsA:
            aTempSum += data.iloc[i, n]

        aSum += aTempSum
        aMax = max(aTempSum, aMax)
        aMin = min(aTempSum, aMax)

        for n in attrsB:
            bTempSum += data.iloc[i, n]
        bSum += bTempSum
        bMax = max(bTempSum, bMax)
        bMin = min(bTempSum, bMin)

        sum += math.pow(aSum - bSum, 2) / (len(attrsA) + len(attrsB))
    sum = math.sqrt(sum)
    maxDifference = max(abs(aMax-bMin), abs(aMax-bMin))

    return [sum, maxDifference]