PYTHON——自然间断点分级法
import xlrd
import xlwt
import jenkspy
Table= xlrd.open_workbook("D:\\dxkj\\test\\因子.xlsx")
print(Table.sheet_names())
sheet1 = Table.sheets()[0]
print(sheet1.name,sheet1.nrows, sheet1.ncols)
print(sheet1.col_values(1))
f = xlwt.Workbook()
for k in range(0,len(Table.sheets())):
sheetSave = f.add_sheet(Table.sheets()[k].name,cell_overwrite_ok = True)
sheet = Table.sheets()[k]
for j in range(1,sheet.ncols):
col = sheet.col_values(j)
List = col[1:len(col)]
newList=col[1:len(col)]
newList.sort()
nn=int(len(col)*0.025)
del newList[0:nn]
del newList[0-nn:]
ListID = col[1:len(col)]
breaks = jenkspy.jenks_breaks(newList,nb_class = 5)
print(breaks)
for i in range(len(List)):
if List[i] >5:
print(List[i] )
if List[i] < breaks[1]:
ListID[i] = 1
elif List[i] < breaks[2] and List[i] >= breaks[1]:
ListID[i] = 2
elif List[i] < breaks[3] and List[i] >= breaks[2]:
ListID[i] = 3
elif List[i] < breaks[4] and List[i] >= breaks[3]:
ListID[i] = 4
elif List[i] >= breaks[4]:
ListID[i] = 5
else:
ListID[i] = 5
for u in range(0,len(ListID)):
sheetSave.write(u,j,ListID[u])
print(j,'------')
print(k,'======')
f.save(r'D:\demo.xls')