import numpy as np
def gini(data_list):
'''获得列表中的gini系数'''
data_length = len(data_list)
total_sum = np.sum(data_list)
total_gini = 0
for i in range(data_length):
temp_denominator = data_list[i][0] + data_list[i][1]
temp_son = np.sum(data_list[i])
temp_gini = 1 - np.power(data_list[i][0]/temp_denominator, 2) - np.power(data_list[i][1]/temp_denominator, 2)
total_gini = temp_gini * temp_son / total_sum + total_gini
return total_gini
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
测试: