自己写的聚类

好久没更新了，最近研究K线去了，今天收到朋友的请求帮助写个聚类的分析，写了半天，50行带码7，8个bug，不过好在分析代码和程序代码的不同是非常明显的，做数据分析的带码，重要的是你一定要能算出结果，而且过程数据要清晰不能出错，有bug没关系主逻辑能跑就行，但是业务代码就不行，除了主逻辑要能正常跑外，非正常逻辑也要涉及到，要不然就会触发用户的bug，因为用户可能不是按照你想象的路子走，很多bug都是给不走寻常路的用户搞出来的。

记录一下今天自己写的聚类，用于分析光谱数据，其实就是二维矩阵分析法，数据相似在同一个范围会聚成一类，用到了pandas和大量的二维数组计算

#-*- coding = utf-8 -*-
import pandas as pd
import math
import numpy as np
import  copy
# def fun_avg(listTemp, n):
#     for i in range(0, len(listTemp), n):
#         yield listTemp[i:i + n]

def fun_avg(items, n):
    return [items[i:i+n] for i in range(0, len(items), n)]

print("请将需要计算的文件名放入当前目录下，只需要第一个表不能有空格，并以js.xlsx命名结尾")

print("请输入您要计算的光波最小值")
fenleiqujian_min = int(input("整数:"))
print("请输入您要计算的光波最大值")
fenleiqujian_max= int(input("整数:"))
print("请输入您想将光波区间分为多少类")
fenleiqujian_num = int(input("整数:"))
print("请输入您想将光波吸收程度分为多少类")
xishoudu_num= int(input("整数:"))

# fenleiqujian_min=557  # 光照分类波长区间最小值
# fenleiqujian_max=589  # 光照分类波长区间最大值
#
# fenleiqujian_num = 6 # 光照分类数量
#
#
# xishoudu_num=3  #吸收度分类数量



df = pd.read_excel("./js.xlsx")
df = df.set_index('No')
min_biaotou = int(min(df.columns.tolist()))
df1 = df.iloc[:,(fenleiqujian_min - min_biaotou +1 ): (fenleiqujian_max - min_biaotou +1) ]




index_a = df1.index.tolist()

column_a = df1.columns.tolist()
data_a = df1.values.tolist()

data_b = []
jiange = math.floor(len(column_a)/fenleiqujian_num)

for array_a in data_a:
    temp_list = fun_avg(array_a,jiange)


    temp_list2 = []
    for i in temp_list:
        temp_list2.append(np.mean(i))
    data_b.append(temp_list2)




max = data_b[0][0]
min = data_b[0][0]
for m in data_b:
    for n in m:
        if float(n) >= float(max):
            max = n
        if float(n) <= float(min):
            min = n

keduzhi = (max - min)/xishoudu_num  #刻度值
if max-min < 0.025:
    keduzhi =1  #刻度值

zulei=[]
for x in data_b:
    temp_y =''
    for y in x:
        zu = int((y-min)/keduzhi)
        temp_y = temp_y+str(zu)
    # print(temp_y)
    zulei.append(temp_y)


zulei1 = copy.deepcopy(zulei)
zhonglei  = list(set(zulei))

zuizhong = list(zip(zulei,index_a))

zuizhong.sort()

file = open('./daan.txt','a+')

i_zhong = 1
for c in zhonglei:
    file.write("第"+str(i_zhong)+'种：')
    i_zhong+=1
    for ii in zuizhong:
        if (ii[0] == c):
            file.write(str(ii[1])+';')
    file.write("\r\n")
file.close()