保存指标图片

import os  
import matplotlib.pyplot as plt  
import numpy as np  
import tensorflow as tf
from scipy.stats import wilcoxon
from scipy.stats import kruskal
from scipy.stats import mannwhitneyu
from scipy.stats import f_oneway
from scipy.stats import ttest_rel
from scipy.stats import ttest_ind
from scipy.stats import chi2_contingency
from scipy.stats import kendalltau  
from scipy.stats import shapiro
from six.moves import xrange  
import math
from scipy.stats.mstats_basic import ttest_1samp
np.set_printoptions(suppress=True)
np.set_printoptions(threshold=np.inf) #输出全部矩阵不带省略号
# 生成数据节点处存在误差，这里函数计算每个分支第一个点的平均值，然后再赋值给第一个点，以便后面计算角度
def manage_data(data):
      data = data.reshape(-1,16,16)
      manage_data = []
      data = data.tolist()
      for i in range(len(data)):
          zhu_x = data[i][0]
          zuo_x = data[i][1]
          you_x = data[i][2]
          zhu_y = data[i][3]
          zuo_y = data[i][4]
          you_y = data[i][5]
          zhu_diam = data[i][6]
          zuo_diam = data[i][7]
          you_diam = data[i][8]
          diam = data[i][9]
          x0_mean = (zhu_x[0] + zuo_x[0] + you_x[0]) / 3.
          y0_mean = (zhu_y[0] + zuo_y[0] + you_y[0]) / 3.
          zhu_x[0] = x0_mean
          zuo_x[0] = x0_mean
          you_x[0] = x0_mean
          zhu_y[0] = y0_mean
          zuo_y[0] = y0_mean
          you_y[0] = y0_mean
          fencha = [zhu_x] + [zuo_x] + [you_x] + [zhu_y] + [zuo_y] + [you_y] + [zhu_diam] + [zuo_diam] + [you_diam] + [diam]
          manage_data.append(fencha)
      manage_data = np.array(manage_data)
      
      return manage_data
def save_imgs(data,path,label):
  #dpi=10,1000x1000 dpi=1,100x100
  for i in range(len(data)):
    zhu_x = data[i][0]
    zuo_x = data[i][1]
    you_x = data[i][2]
    zhu_y = data[i][3]
    zuo_y = data[i][4]
    you_y = data[i][5]
    zhu_diam = data[i][6]
    zuo_diam = data[i][7]
    you_diam= data[i][8]
    fig = plt.figure(figsize=(100,100))
    plt.plot(zhu_x,zhu_y,color='red',linewidth=400*zhu_diam[0])
    plt.plot(zuo_x,zuo_y,color='green',linewidth=400*zuo_diam[0])
    plt.plot(you_x,you_y,color='blue',linewidth=400*you_diam[0])
    plt.xlim(0.,1.)
    plt.ylim(0.,1.)
    plt.xticks(np.arange(0.,1.,0.1))
    plt.yticks(np.arange(0.,1.,0.1))
    fig.set_size_inches(100,100)
    plt.axis("off")
    plt.savefig('C:\\Users\\Administrator\\Desktop\\%s\\%s_%d.jpg' % (path,label,i),dpi=10)
    plt.close()
#     plt.show()
    
# model57,seed(4),1000个样本可以
ori_data = np.load('data/分开分叉与汇聚插值16x16.npy')
# ori_data = np.load('data/model9.npy')
ori_data = ori_data.reshape(-1,16,16)
gen_data = np.load('data/model57.npy')
gen_data = gen_data.reshape(-1,16,16)
np.random.seed(4)
select_part = "zhu"
np.random.shuffle(ori_data)
np.random.shuffle(gen_data)
ori_data = ori_data[0:500,:,:] #选择保存多少数据
gen_data = gen_data[0:500,:,:] #选择保存多少数据
ori_data = manage_data(ori_data)
gen_data = manage_data(gen_data)
# save_imgs(gen_data,"gen","gen")
# save_imgs(ori_data,"ori","ori")
####################################################
# #绘图，观察是否正确
# gen = ori_data
# for i in range(len(gen)):
#     zhu_x = gen[i][0]
#     zuo_x = gen[i][1]
#     you_x = gen[i][2]
#     zhu_y = gen[i][3]
#     zuo_y = gen[i][4]
#     you_y = gen[i][5]
#     zhu_diam = gen[i][6]
#     zuo_diam = gen[i][7]
#     you_diam = gen[i][8]
#     plt.plot(zhu_x, zhu_y, color="red")
#     plt.plot(zuo_x, zuo_y, color="green")
#     plt.plot(you_x, you_y, color="blue")
#     plt.xlim(0,1)
#     plt.ylim(0,1)
#     plt.xticks(np.arange(0,1,0.1))
#     plt.yticks(np.arange(0,1,0.1))
#     plt.show()
##################################################
# 距离计算公式
def get_len(x1,x2,y1,y2):
    diff_x = (x1-x2)**2
    diff_y = (y1-y2)**2
    length = np.sqrt(diff_x+diff_y)
    return length
  
# 余弦定理计算角度公式,c夹在a,b中间
def cal_angle(a,b,c):
    cos_angle = (a**2+b**2-c**2)/(2*a*b)
    angle = np.arccos(cos_angle)
    angle = math.degrees(angle)
    return angle
  
# 得到管径值,管径在第一个值
def get_diam(fencha):
  zhu_diam = fencha[6][0]
  zuo_diam = fencha[7][0]
  you_diam = fencha[8][0]
  
  return zhu_diam, zuo_diam, you_diam

# 计算端点长度
def get_duandian_len(zhu_x,zhu_y,zuo_x,zuo_y,you_x,you_y):
    #主分支头尾坐标
    zhu_x_tou = zhu_x[0]
    zhu_y_tou = zhu_y[0]
    zhu_x_wei = zhu_x[-1]
    zhu_y_wei = zhu_y[-1]
    #左分支头尾坐标
    zuo_x_tou = zuo_x[0]
    zuo_y_tou = zuo_y[0]
    zuo_x_wei = zuo_x[-1]
    zuo_y_wei = zuo_y[-1]
    #右分支头尾坐标
    you_x_tou = you_x[0]
    you_y_tou = you_y[0]
    you_x_wei = you_x[-1]
    you_y_wei = you_y[-1]
    #主分支端点长
    zhu_duan_len = get_len(zhu_x_tou,zhu_x_wei,zhu_y_tou,zhu_y_wei)
    #左分支端点长
    zuo_duan_len = get_len(zuo_x_tou,zuo_x_wei,zuo_y_tou,zuo_y_wei)
    #右分支端点长
    you_duan_len = get_len(you_x_tou,you_x_wei,you_y_tou,you_y_wei)
    return zhu_duan_len,zuo_duan_len,you_duan_len
  
#计算分支总长度
def get_total_len(zhu_x,zhu_y,zuo_x,zuo_y,you_x,you_y):
    zhu_lin_list = []
    zuo_lin_list = []
    you_lin_list = []
    for i in range(1,len(zhu_x)):
        zhu_lin_len = get_len(zhu_x[i-1],zhu_x[i],zhu_y[i-1],zhu_y[i])
        zhu_lin_list.append(zhu_lin_len)
    for i in range(1,len(zuo_x)):
        zuo_lin_len = get_len(zuo_x[i-1],zuo_x[i],zuo_y[i-1],zuo_y[i])
        zuo_lin_list.append(zuo_lin_len)
    for i in range(1, len(you_x)):
        you_lin_len = get_len(you_x[i-1],you_x[i],you_y[i-1],you_y[i])
        you_lin_list.append(you_lin_len)
    zhu_total_len = 0
    for file in zhu_lin_list:
        zhu_total_len += file
    zuo_total_len = 0
    for file in zuo_lin_list:
        zuo_total_len += file
    you_total_len = 0
    for file in you_lin_list:
        you_total_len += file
    return zhu_total_len,zuo_total_len,you_total_len
# #############################################################
# #测试函数正确性
# zhu_x = [0,1,2,3,3,3,3,3] #理论上端点长5，总长度7,本身是一个两条直角边为3，4的直角三角形
# zhu_y = [0,0,0,0,1,2,3,4]
# zhu_duan_len,zuo_duan_len,you_duan_len = get_duandian_len(zhu_x,zhu_y,zhu_x,zhu_y,zhu_x,zhu_y)
# zhu_total_len,zuo_total_len,you_total_len = get_total_len(zhu_x,zhu_y,zhu_x,zhu_y,zhu_x,zhu_y)
# print(zhu_duan_len,zuo_total_len)
# print(zhu_total_len,zuo_total_len)
# ################################################################
# 计算卷曲度
def get_juanqu(zhu_duan_len,zuo_duan_len,you_duan_len,zhu_total_len,zuo_total_len,you_total_len):
    zhu_juanqu = zhu_total_len / zhu_duan_len
    zuo_juanqu = zuo_total_len / zuo_duan_len
    you_juanqu = you_total_len / you_duan_len
#     print(zhu_total_len)
#     print(zhu_duan_len)
#     print(zhu_juanqu)
    return zhu_juanqu,zuo_juanqu,you_juanqu
  
# 计算角度
def get_angle(zhu_x, zhu_y, zuo_x, zuo_y, you_x, you_y):
  # 假设节点处的点都为a,zhu_a,zuo_a,you_a它们x,y坐标点都相等，然后
  # 后面一个点都为b,zhu_b,zuo_b,you_b,它们的值不相等
  # zhu_zuo_angle = zhu_ab,zuo_ab,zhu_zuo_b
  # zhu_you_angle = zhu_ab,you_ab,zhu_you_b
  # zuo_you_angle = zuo_ab,you_ab,zuo_you_b 
  # 主分支上两个点
  zhu_x_a = zhu_x[0]
  zhu_y_a = zhu_y[0]
  zhu_x_b = zhu_x[1]
  zhu_y_b = zhu_y[1]
  # 左分支上两个点
  zuo_x_a = zuo_x[0]
  zuo_y_a = zuo_y[0]
  zuo_x_b = zuo_x[1]
  zuo_y_b = zuo_y[1]
  # 右分支上两个点
  you_x_a = you_x[0]
  you_y_a = you_y[0]
  you_x_b = you_x[1]
  you_y_b = you_y[1]
  # zhu_x_a, zuo_x_a, you_x_a应该相等
  # 每个分支端点长度
  zhu_ab_len = get_len(zhu_x_a,zhu_x_b,zhu_y_a,zhu_y_b)
  zuo_ab_len = get_len(zuo_x_a,zuo_x_b,zuo_y_a,zuo_y_b)
  you_ab_len = get_len(you_x_a,you_x_b,you_y_a,you_y_b)
  zhu_zuo_len = get_len(zhu_x_b,zuo_x_b,zhu_y_b,zuo_y_b)
  zhu_you_len = get_len(zhu_x_b,you_x_b,zhu_y_b,you_y_b)
  zuo_you_len = get_len(zuo_x_b,you_x_b,zuo_y_b,you_y_b)
  zhu_zuo_angle = cal_angle(zhu_ab_len,zuo_ab_len,zhu_zuo_len)
  zhu_you_angle = cal_angle(zhu_ab_len,you_ab_len,zhu_you_len)
  zuo_you_angle = cal_angle(zuo_ab_len,you_ab_len,zuo_you_len)
  
  # 三个角度有可能其中为nan,加在一块zong_angle也就是nan,在这里不做处理后面再进行处理
  zong_angle = zhu_zuo_angle + zhu_you_angle + zuo_you_angle
  
  return zhu_zuo_angle,zhu_you_angle,zuo_you_angle,zong_angle

# 得到管径比
def get_diam_ratio(zhu_diam, zuo_diam, you_diam):
  # 管径可能为0
  if zuo_diam == 0:
    zuo_diam = 1
  if you_diam == 0:
    you_diam = 1
  zhu_zuo_diam_ratio = zhu_diam / zuo_diam
  zhu_you_diam_ratio = zhu_diam / you_diam
  zuo_you_diam_ratio = zuo_diam / you_diam
  return  zhu_zuo_diam_ratio, zhu_you_diam_ratio, zuo_you_diam_ratio

# 得到长度比
def get_len_ratio(zhu_total_len, zuo_total_len, you_total_len):
  zhu_zuo_len_ratio = zhu_total_len / zuo_total_len
  zhu_you_len_ratio = zhu_total_len / you_total_len
  zuo_you_len_ratio = zuo_total_len / you_total_len
  
  return zhu_zuo_len_ratio, zhu_you_len_ratio, zuo_you_len_ratio

# 打印数据
def printList(data_list):
  data_list = sorted(data_list)
#   data_list = np.array(data_list)
  print(data_list)
    
# part显示测哪一部分,如果为zhu,则测主分支的各项指标.(zhu,zuo,you,all)
def calculate(data, part="zhu"):
    # 管径列表
    zhu_diam_list = []
    zuo_diam_list = []
    you_diam_list = []
    # 端点长度列表
    zhu_duan_len_list = []
    zuo_duan_len_list = []
    you_duan_len_list = []
    # 总长度列表
    zhu_total_len_list = []
    zuo_total_len_list = []
    you_total_len_list = []
    # 卷曲度列表
    zhu_juanqu_list = []
    zuo_juanqu_list  = []
    you_juanqu_list = []
    # 角度列表  zhu代表zhu_zuo,you代表zhu_you,zuo代表zuo_you
    zhu_zuo_angle_list = []
    zhu_you_angle_list = []
    zuo_you_angle_list = []
    # 管径比列表
    zhu_zuo_diam_ratio_list = []
    zhu_you_diam_ratio_list = []
    zuo_you_diam_ratio_list = []
    # 长度比列表
    zhu_zuo_len_ratio_list = []
    zhu_you_len_ratio_list = []
    zuo_you_len_ratio_list = []
    
    for i in range(0,len(data)):
        zhu_x = data[i][0]
        zuo_x = data[i][1]
        you_x = data[i][2]
        zhu_y = data[i][3]
        zuo_y = data[i][4]
        you_y = data[i][5]
        zhu_diam = data[i][6]
        zuo_diam = data[i][7]
        you_diam = data[i][8]
        
          
        # 计算每个分支管径
        # data[i]代表一个分叉矩阵10x10
        zhu_diam, zuo_diam, you_diam = get_diam(data[i])
        zhu_diam_list.append((zhu_diam,i))
        zuo_diam_list.append((zuo_diam,i))
        you_diam_list.append((you_diam,i))
        
        # 计算每个分叉分支端点长度
        zhu_duan_len,zuo_duan_len,you_duan_len = get_duandian_len(zhu_x,zhu_y,zuo_x,zuo_y,you_x,you_y)
        zhu_duan_len_list.append((zhu_duan_len,i))
        zuo_duan_len_list.append((zuo_duan_len,i))
        you_duan_len_list.append((you_duan_len,i))
        
        # 计算每个分叉分支总长度
        zhu_total_len,zuo_total_len,you_total_len = get_total_len(zhu_x,zhu_y,zuo_x,zuo_y,you_x,you_y)
        zhu_total_len_list.append((zhu_total_len,i))
        zuo_total_len_list.append((zuo_total_len,i))
        you_total_len_list.append((you_total_len,i))
        
        # 计算每个分叉分支卷曲度
        zhu_juanqu,zuo_juanqu,you_juanqu = get_juanqu(zhu_duan_len,zuo_duan_len,you_duan_len,zhu_total_len,zuo_total_len,you_total_len)
        zhu_juanqu_list.append((zhu_juanqu,i))
        zuo_juanqu_list.append((zuo_juanqu,i))
        you_juanqu_list.append((you_juanqu,i))
        
        
        # 计算每个分叉分支管径比
        zhu_zuo_diam_ratio, zhu_you_diam_ratio, zuo_you_diam_ratio = get_diam_ratio(zhu_diam, zuo_diam, you_diam)
        zhu_zuo_diam_ratio_list.append((zhu_zuo_diam_ratio,i))
        zhu_you_diam_ratio_list.append((zhu_you_diam_ratio,i))
        zuo_you_diam_ratio_list.append((zuo_you_diam_ratio,i))
        
        # 计算每个分叉分支长度比
        zhu_zuo_len_ratio, zhu_you_len_ratio, zuo_you_len_ratio = get_len_ratio(zhu_total_len, zuo_total_len, you_total_len)
        zhu_zuo_len_ratio_list.append((zhu_zuo_len_ratio,i))
        zhu_you_len_ratio_list.append((zhu_you_len_ratio,i))
        zuo_you_len_ratio_list.append((zuo_you_len_ratio,i))
        
        # 计算每个分叉分支角度
        zhu_zuo_angle,zhu_you_angle,zuo_you_angle,zong_angle = get_angle(zhu_x, zhu_y, zuo_x, zuo_y, you_x, you_y)
        if math.isnan(zong_angle):
          continue
        else:
          zhu_zuo_angle_list.append((zhu_zuo_angle,i))
          zhu_you_angle_list.append((zhu_you_angle,i))
          zuo_you_angle_list.append((zuo_you_angle,i))
    # 一共七项指标
    # 混合管径值
    diam_list = zhu_diam_list + zuo_diam_list + you_diam_list
    # 混合端点长度
    duan_len_list = zhu_duan_len_list + zuo_duan_len_list + you_duan_len_list
    # 混合总长度
    total_len_list = zhu_total_len_list + zuo_total_len_list + you_total_len_list
    # 混合卷曲度
    juanqu_list = zhu_juanqu_list + zuo_juanqu_list + you_juanqu_list
    # 混合角度
    angle_list = zhu_zuo_angle_list + zhu_you_angle_list + zuo_you_angle_list
    # 混合管径比
    diam_ratio_list = zhu_zuo_diam_ratio_list + zhu_you_diam_ratio_list + zuo_you_diam_ratio_list
    # 混合长度比
    len_ratio_list = zhu_zuo_len_ratio_list + zhu_you_len_ratio_list + zuo_you_len_ratio_list
    
#     print("**********************************************************************")
#     printList(zuo_you_len_ratio_list)
    
    # 分别返回各项指标
    if part == "zhu":
      return zhu_diam_list, zhu_duan_len_list, zhu_total_len_list, zhu_juanqu_list,zhu_zuo_angle_list,zhu_zuo_diam_ratio_list,zhu_zuo_len_ratio_list
    if part == "zuo":
      return zuo_diam_list, zuo_duan_len_list, zuo_total_len_list, zuo_juanqu_list,zhu_you_angle_list,zhu_you_diam_ratio_list,zhu_you_len_ratio_list
    if part == "you":
      return you_diam_list, you_duan_len_list, you_total_len_list, you_juanqu_list,zuo_you_angle_list,zuo_you_diam_ratio_list,zuo_you_len_ratio_list
    if part == "all":
      return diam_list, duan_len_list, total_len_list, juanqu_list,angle_list,diam_ratio_list,len_ratio_list
    
    
#通过改变part,来分别计算每个分支的指标
ori_diam_list, ori_duan_len_list, ori_total_len_list, ori_juanqu_list, ori_angle_list, ori_diam_ratio_list, ori_len_ratio_list = calculate(ori_data,part=select_part)
gen_diam_list, gen_duan_len_list, gen_total_len_list, gen_juanqu_list, gen_angle_list, gen_diam_ratio_list, gen_len_ratio_list = calculate(gen_data,part=select_part)
def show_imgs(data,data_list,model):
  # 对指标进行排序，然后保存
  data_list = sorted(data_list)
  for i in range(len(data_list)):
      metrics = data_list[i][0]
      print(metrics)
      index = data_list[i][1]
      zhu_x = data[index][0]
      zuo_x = data[index][1]
      you_x = data[index][2]
      zhu_y = data[index][3]
      zuo_y = data[index][4]
      you_y = data[index][5]
      zhu_diam = data[index][6]
      zuo_diam = data[index][7]
      you_diam = data[index][8]
      fig = plt.figure(figsize=(100,100))
      plt.plot(zhu_x,zhu_y,color='red',linewidth=400*zhu_diam[0])
      plt.plot(zuo_x,zuo_y,color='green',linewidth=400*zuo_diam[0])
      plt.plot(you_x,you_y,color='blue',linewidth=400*you_diam[0])
      plt.xlim(0.,1.)
      plt.ylim(0.,1.)
      plt.xticks(np.arange(0.,1.,0.1))
      plt.yticks(np.arange(0.,1.,0.1))
      fig.set_size_inches(100,100)
      plt.axis("off")
      plt.savefig('C:\\Users\\Administrator\\Desktop\\可视化各指标分叉\\主左长度比\\gen\\%s_%.5f.jpg' % (model,metrics),dpi=10)
      plt.close()
show_imgs(gen_data,gen_diam_ratio_list,'gen')
# show_imgs(ori_data,ori_len_ratio_list,'ori')

##############################################################################
# print(len(ori_diam_list),len(gen_diam_list))
# print(len(ori_duan_len_list),len(gen_duan_len_list))
# print(len(ori_juanqu_list),len(gen_juanqu_list))
# print(len(ori_diam_list),len(gen_diam_list))
# print(len(ori_diam_ratio_list),len(gen_diam_ratio_list))
# print(len(ori_angle_list),len(gen_angle_list))
'''
###############################################################################
def calculate_param(data_diam_list, data_duan_len_list, data_total_len_list, data_juanqu_list, data_angle_list, data_diam_ratio_list, data_len_ratio_list):
    #计算各项指标平均值
    data_diam_list_mean = np.mean(data_diam_list)
    data_duan_len_list_mean = np.mean(data_duan_len_list)
    data_total_len_list_mean = np.mean(data_total_len_list)
    data_juanqu_list_mean = np.mean(data_juanqu_list)
    data_angle_list_mean = np.mean(data_angle_list)
    data_diam_ratio_list_mean = np.mean(data_diam_ratio_list)
    data_len_ratio_list_mean = np.mean(data_len_ratio_list)
    
    #计算各项指标方差
    data_diam_list_var = np.var(data_diam_list)
    data_duan_len_list_var = np.var(data_duan_len_list)
    data_total_len_list_var = np.var(data_total_len_list)
    data_juanqu_list_var = np.var(data_juanqu_list)
    data_angle_list_var = np.var(data_angle_list)
    data_diam_ratio_list_var = np.var(data_diam_ratio_list)
    data_len_ratio_list_var = np.var(data_len_ratio_list)
    
    #计算各项指标标准差
    data_diam_list_std = np.std(data_diam_list,ddof=1)
    data_duan_len_list_std = np.std(data_duan_len_list,ddof=1)
    data_total_len_list_std = np.std(data_total_len_list,ddof=1)
    data_juanqu_list_std = np.std(data_juanqu_list,ddof=1)
    data_angle_list_std = np.std(data_angle_list,ddof=1)
    data_diam_ratio_list_std = np.std(data_diam_ratio_list,ddof=1)
    data_len_ratio_list_std = np.std(data_len_ratio_list,ddof=1)
       
    return data_diam_list_mean, data_duan_len_list_mean, data_total_len_list_mean, data_juanqu_list_mean, data_angle_list_mean, data_diam_ratio_list_mean, data_len_ratio_list_mean,\
    data_diam_list_var, data_duan_len_list_var, data_total_len_list_var, data_juanqu_list_var, data_angle_list_var, data_diam_ratio_list_var, data_len_ratio_list_var,\
    data_diam_list_std, data_duan_len_list_std, data_total_len_list_std, data_juanqu_list_std, data_angle_list_std, data_diam_ratio_list_std, data_len_ratio_list_std
def culculate_diam_list(ori_diam_list,gen_diam_list):
    # 计算管径是否符合
    # T检验
    ttest_diam_stat,ttest_diam_p = ttest_ind(ori_diam_list,gen_diam_list)
    # 曼-惠特尼U检验
    man_diam_stat, man_diam_p = mannwhitneyu(ori_diam_list,gen_diam_list)
    # 威尔科克森符号秩检验
    wil_diam_stat, wil_diam_p = wilcoxon(ori_diam_list,gen_diam_list)
    
    return ttest_diam_stat, ttest_diam_p, man_diam_stat, man_diam_p, wil_diam_stat, wil_diam_p
       
def calculate_duan_len_test(ori_duan_len_list,gen_duan_len_list):
    #计算端点长度是否符合
    #T检验
    ttest_duan_len_stat,ttest_duan_len_p = ttest_ind(ori_duan_len_list,gen_duan_len_list)
    #曼-惠特尼U检验
    man_duan_len_stat,man_duan_len_p = mannwhitneyu(ori_duan_len_list,gen_duan_len_list)
    #威尔科克森符号秩检验
    wil_duan_len_stat,wil_duan_len_p = wilcoxon(ori_duan_len_list,gen_duan_len_list)
    
    return ttest_duan_len_stat, ttest_duan_len_p, man_duan_len_stat, man_duan_len_p, wil_duan_len_stat, wil_duan_len_p
       
def calculate_total_len_test(ori_total_len_list,gen_total_len_list):
    #计算总长度是否符合
    #T检验
    ttest_total_len_stat,ttest_total_len_p = ttest_ind(ori_total_len_list,gen_total_len_list)
    #曼-惠特尼U检验
    man_total_len_stat,man_total_len_p = mannwhitneyu(ori_total_len_list,gen_total_len_list)
    #威尔科克森符号秩检验
    wil_total_len_stat,wil_total_len_p = wilcoxon(ori_total_len_list,gen_total_len_list)
    return ttest_total_len_stat, ttest_total_len_p, man_total_len_stat, man_total_len_p, wil_total_len_stat, wil_total_len_p
   
   
def calculate_juanqu_test(ori_juanqu_list,gen_juanqu_list):
    #计算卷曲度是否符合
    #T检验
    ttest_juanqu_stat,ttest_juanqu_p = ttest_ind(ori_juanqu_list,gen_juanqu_list)
    #曼-惠特尼U检验
    man_juanqu_stat,man_juanqu_p = mannwhitneyu(ori_juanqu_list,gen_juanqu_list)
    #威尔科克森符号秩检验
    wil_juanqu_stat,wil_juanqu_p = wilcoxon(ori_juanqu_list,gen_juanqu_list)
    
    return ttest_juanqu_stat, ttest_juanqu_p, man_juanqu_stat, man_juanqu_p, wil_juanqu_stat, wil_juanqu_p

def calculate_angle_test(ori_angle_list,gen_angle_list):
    # 使角度列表长度相等，因为之前角度为nan时直接跳出循环了
    if len(ori_angle_list) > len(gen_angle_list):
      ori_angle_list = ori_angle_list[0:len(gen_angle_list)]
    else:
      gen_angle_list = gen_angle_list[0:len(ori_angle_list)]
    # 计算角度是否符合
    # T检验
    ttest_angle_stat,ttest_angle_p = ttest_ind(ori_angle_list,gen_angle_list)
    # 曼-惠特尼U检验
    man_angle_stat,man_angle_p = mannwhitneyu(ori_angle_list,gen_angle_list)
    # 威尔科克森符号秩检验
    wil_angle_stat,wil_angle_p = wilcoxon(ori_angle_list,gen_angle_list)
    
    return ttest_angle_stat, ttest_angle_p, man_angle_stat, man_angle_p, wil_angle_stat, wil_angle_p
  
def calculate_diam_ratio_test(ori_diam_ratio_list,gen_diam_ratio_list):
    # 计算管径比是否符合
    # T检验
    ttest_diam_ratio_stat,ttest_diam_ratio_p = ttest_ind(ori_diam_ratio_list,gen_diam_ratio_list)
    # 曼-惠特尼U检验
    man_diam_ratio_stat,man_diam_ratio_p = mannwhitneyu(ori_diam_ratio_list,gen_diam_ratio_list)
    # 威尔科克森符号秩检验
    wil_diam_ratio_stat,wil_diam_ratio_p = wilcoxon(ori_diam_ratio_list,gen_diam_ratio_list)
    
    return ttest_diam_ratio_stat, ttest_diam_ratio_p, man_diam_ratio_stat, man_diam_ratio_p, wil_diam_ratio_stat, wil_diam_ratio_p
  
def calculate_len_ratio_test(ori_len_ratio_list,gen_len_ratio_list):
    # 计算管径比是否符合
    # T检验
    ttest_len_ratio_stat,ttest_len_ratio_p = ttest_ind(ori_len_ratio_list,gen_len_ratio_list)
    # 曼-惠特尼U检验
    man_len_ratio_stat,man_len_ratio_p = mannwhitneyu(ori_len_ratio_list,gen_len_ratio_list)
    # 威尔科克森符号秩检验
    wil_len_ratio_stat,wil_len_ratio_p = wilcoxon(ori_len_ratio_list,gen_len_ratio_list)
    
    return ttest_len_ratio_stat, ttest_len_ratio_p, man_len_ratio_stat, man_len_ratio_p, wil_len_ratio_stat, wil_len_ratio_p     
       
#计算原始数据与生成数据各项指标的平均值，方差，标准差
ori_diam_list_mean,ori_duan_len_list_mean,ori_total_len_list_mean,ori_juanqu_list_mean,ori_angle_list_mean, ori_diam_ratio_list_mean, ori_len_ratio_list_mean,ori_diam_list_var,ori_duan_len_list_var,ori_total_len_list_var,ori_juanqu_list_var,ori_angle_list_var, ori_diam_ratio_list_var, ori_len_ratio_list_var,ori_diam_list_std,ori_duan_len_list_std,ori_total_len_list_std,ori_juanqu_list_std ,ori_angle_list_std, ori_diam_ratio_list_std, ori_len_ratio_list_std = \
calculate_param(ori_diam_list,ori_duan_len_list,ori_total_len_list,ori_juanqu_list, ori_angle_list, ori_diam_ratio_list, ori_len_ratio_list)

gen_diam_list_mean,gen_duan_len_list_mean,gen_total_len_list_mean,gen_juanqu_list_mean,gen_angle_list_mean, gen_diam_ratio_list_mean, gen_len_ratio_list_mean,gen_diam_list_var,gen_duan_len_list_var,gen_total_len_list_var,gen_juanqu_list_var,gen_angle_list_var, gen_diam_ratio_list_var, gen_len_ratio_list_var,gen_diam_list_std, gen_duan_len_list_std,gen_total_len_list_std,gen_juanqu_list_std ,gen_angle_list_std, gen_diam_ratio_list_std, gen_len_ratio_list_std = \
calculate_param(gen_diam_list,gen_duan_len_list,gen_total_len_list,gen_juanqu_list, gen_angle_list, gen_diam_ratio_list, gen_len_ratio_list)
print("----------管径----------")
print("原始数据平均值:%f" % ori_diam_list_mean,"生成数据平均值:%f" % gen_diam_list_mean)
print("原始数据方差:%f" % ori_diam_list_var,"生成数据方差:%f" % gen_diam_list_var)
print("原始数据标准差:%f" % ori_diam_list_std,"生成数据标准差:%f" % gen_diam_list_std)
print("\n")
print("----------端点长度----------")
print("原始数据平均值:%f" % ori_duan_len_list_mean,"生成数据平均值:%f" % gen_duan_len_list_mean)
print("原始数据方差:%f" % ori_duan_len_list_var,"生成数据方差:%f" % gen_duan_len_list_var)
print("原始数据标准差:%f" % ori_duan_len_list_std,"生成数据标准差:%f" % gen_duan_len_list_std)
print("\n")
print("----------总长度----------")
print("原始数据平均值:%f" % ori_total_len_list_mean,"生成数据平均值:%f" % gen_total_len_list_mean)
print("原始数据方差:%f" % ori_total_len_list_var,"生成数据方差:%f" % gen_total_len_list_var)
print("原始数据标准差:%f" % ori_total_len_list_std,"生成数据标准差:%f" % gen_total_len_list_std)
print("\n")
print("----------卷曲度----------")
print("原始数据平均值:%f" % ori_juanqu_list_mean,"生成数据平均值:%f" % gen_juanqu_list_mean)
print("原始数据方差:%f" % ori_juanqu_list_var,"生成数据标准差:%f" % gen_juanqu_list_var)
print("原始数据标准差:%f" % ori_juanqu_list_std,"生成数据标准差:%f" % gen_juanqu_list_std)
print("\n")
print("----------角度----------")
print("原始数据平均值:%f" % ori_angle_list_mean,"生成数据平均值:%f" % gen_angle_list_mean)
print("原始数据方差:%f" % ori_angle_list_var,"生成数据标准差:%f" % gen_angle_list_var)
print("原始数据标准差:%f" % ori_angle_list_std,"生成数据标准差:%f" % gen_angle_list_std)
print("\n")
print("----------管径比----------")
print("原始数据平均值:%f" % ori_diam_ratio_list_mean,"生成数据平均值:%f" % gen_diam_ratio_list_mean)
print("原始数据方差:%f" % ori_diam_ratio_list_var,"生成数据标准差:%f" % gen_diam_ratio_list_var)
print("原始数据标准差:%f" % ori_diam_ratio_list_std,"生成数据标准差:%f" % gen_diam_ratio_list_std)
print("\n")
print("----------长度比----------")
print("原始数据平均值:%f" % ori_len_ratio_list_mean,"生成数据平均值:%f" % gen_len_ratio_list_mean)
print("原始数据方差:%f" % ori_len_ratio_list_var,"生成数据标准差:%f" % gen_len_ratio_list_var)
print("原始数据标准差:%f" % ori_len_ratio_list_std,"生成数据标准差:%f" % gen_len_ratio_list_std)
  
print("\n\n")
# 计算原始数据与生成数据是否符合各项统计性检验
# 各项指标正态性检验都不符合, 这里不再打印
ttest_diam_stat, ttest_diam_p, man_diam_stat, man_diam_p, wil_diam_stat, wil_diam_p= culculate_diam_list(ori_diam_list,gen_diam_list)
print("----------管径统计性检验----------")
print("T检验:%f" % ttest_diam_p)
print("曼-惠特尼U检验:%f" % man_diam_p)
print("威尔科克森符号秩检验:%f" % wil_diam_p)
print("\n")
ttest_duan_len_stat, ttest_duan_len_p, man_duan_len_stat, man_duan_len_p, wil_duan_len_stat, wil_duan_len_p = calculate_duan_len_test(ori_duan_len_list,gen_duan_len_list)
print("----------端点长度统计性检验----------")
print("T检验:%f" % ttest_duan_len_p)
print("曼-惠特尼U检验:%f" % man_duan_len_p)
print("威尔科克森符号秩检验:%f" % wil_duan_len_p)
print("\n")
ttest_total_len_stat, ttest_total_len_p, man_total_len_stat, man_total_len_p, wil_total_len_stat, wil_total_len_p = calculate_total_len_test(ori_total_len_list,gen_total_len_list)
print("----------总长度统计性检验----------")
print("T检验:%f" % ttest_total_len_p)
print("曼-惠特尼U检验:%f" % man_total_len_p)
print("威尔科克森符号秩检验:%f" % wil_total_len_p)
print("\n")
ttest_juanqu_stat, ttest_juanqu_p, man_juanqu_stat, man_juanqu_p, wil_juanqu_stat, wil_juanqu_p = calculate_juanqu_test(ori_juanqu_list,gen_juanqu_list)
print("----------卷曲度统计性检验----------")
print("T检验:%f" % ttest_juanqu_p)
print("曼-惠特尼U检验:%f" % man_juanqu_p)
print("威尔科克森符号秩检验:%f" % wil_juanqu_p)
print("\n")
ttest_angle_stat, ttest_angle_p, man_angle_stat, man_angle_p, wil_angle_stat, wil_angle_p = calculate_angle_test(ori_angle_list,gen_angle_list)
print("----------角度统计性检验----------")
print("T检验:%f" % ttest_angle_p)
print("曼-惠特尼U检验:%f" % man_angle_p)
print("威尔科克森符号秩检验:%f" % wil_angle_p)
print("\n")
ttest_diam_ratio_stat, ttest_diam_ratio_p, man_diam_ratio_stat, man_diam_ratio_p, wil_diam_ratio_stat, wil_diam_ratio_p = calculate_diam_ratio_test(ori_diam_ratio_list,gen_diam_ratio_list)
print("----------管径比统计性检验----------")
print("T检验:%f" % ttest_diam_ratio_p)
print("曼-惠特尼U检验:%f" % man_diam_ratio_p)
print("威尔科克森符号秩检验:%f" % wil_diam_ratio_p)
print("\n")
ttest_len_ratio_stat, ttest_len_ratio_p, man_len_ratio_stat, man_len_ratio_p, wil_len_ratio_stat, wil_len_ratio_p = calculate_len_ratio_test(ori_len_ratio_list,gen_len_ratio_list)
print("----------长度比统计性检验----------")
print("T检验:%f" % ttest_len_ratio_p)
print("曼-惠特尼U检验:%f" % man_len_ratio_p)
print("威尔科克森符号秩检验:%f" % wil_len_ratio_p)
print("\n")
'''