python 2组list绘制拟合曲线、计算拟合方程 R^2

python 复制代码
import matplotlib.pyplot as plt
import numpy as np
# plt.rcParams['font.family'] = 'SimHei' # 指定使用中文字体,例如宋体(SimHei)

def Curve_Fitting(x, y, deg):
    parameter = np.polyfit(x, y, deg)    #拟合deg次多项式
    p = np.poly1d(parameter)             #拟合deg次多项式
    aa=''                               #方程拼接  ------------------------------------------------------
    for i in range(deg+1): 
        bb=round(parameter[i], 2)
        if bb>0:
            if i==0:
                bb=str(bb)
            else:
                bb='+'+str(bb)
        else:
            bb=str(bb)
        if deg==i:
            aa=aa+bb
        else:
            aa=aa+bb+'x^'+str(deg-i)    #方程拼接  ------------------------------------------------------
    # 利用相关系数矩阵计算R方
    r2 = round(np.corrcoef(y, p(x))[0,1]**2,2)
    plt.scatter(x, y, label="Comparison", color='b', marker='o')     #原始数据散点图
    plt.plot(x, p(x), color='r')  # 画拟合曲线
   # plt.text(-1,0,aa,fontdict={'size':'10','color':'b'})
    plt.legend([f"R^2: {r2}", f"Fitting Equation: {aa}"])   #拼接好的方程和R方放到图例
    # 添加标签和标题
    plt.xlabel("Old Normalized_TMB Values")
    plt.ylabel("New Normalized_TMB Values")
    plt.title("Comparison of Normalized_TMB")
    plt.savefig('./tmp.jpg')
    plt.show()
    print('曲线方程为:',aa)
    
    print('     r^2为:', r2)


def plot_scatter(old_list, new_list):
    # 示例数据:相同样本的两次运行结果
    # sample_names = ["Sample 1", "Sample 2", "Sample 3", "Sample 4"]
    run1_values = old_list  # 第一次运行的数据
    run2_values = new_list  # 第二次运行的数据
    # 创建散点图
    plt.scatter(run1_values, run2_values, label="Comparison", color='b', marker='o')
    # 添加标签和标题
    plt.xlabel("Old Normalized_TMB Values")
    plt.ylabel("New Normalized_TMB Values")
    plt.title("Comparison of Normalized_TMB")
    # 添加数据点的标签
    # for i, sample_name in enumerate(sample_names):
    #     plt.annotate(sample_name, (run1_values[i], run2_values[i]))
    # 添加一条线表示一对一的关系
    plt.plot([min(run1_values), max(run1_values)], [min(run1_values), max(run1_values)], linestyle='--', color='r', label="1:1 Line")
    # 显示图例
    plt.legend()
    plt.savefig('./tmp.jpg')
    # 显示散点图
    plt.show()





normalzed_tmb_compare_file = 'normalzed_tmb.txt'
old_FFPE_list, new_FFPE_list = [], []
old_CF_list, new_CF_list = [], []
with open(normalzed_tmb_compare_file) as f:
    next(f)
    for line in f:
        line_list = line.strip('\n').split('\t')
        # if float(line_list[0]) > 20:
        #     continue
        old, type_, new = line_list
        if type_ == 'FFPE':
            old_FFPE_list.append(float(line_list[0]))
            new_FFPE_list.append(float(line_list[-1]))
        if type_ == 'CF':
            old_CF_list.append(float(line_list[0]))
            new_CF_list.append(float(line_list[-1])) 
    # Curve_Fitting(old_FFPE_list, new_FFPE_list, 1)
    # Curve_Fitting(old_CF_list, new_CF_list, 1)
    Curve_Fitting(old_CF_list+old_FFPE_list, new_CF_list+new_FFPE_list, 1)
    # plot_scatter(old_list, new_list)
相关推荐
fsnine11 小时前
Python图形化界面——pyqt5教程
开发语言·python·qt
扶尔魔ocy12 小时前
python程序打包成win的exe应用(以OCR应用为例)
python·ocr·中文识别
Psycho_MrZhang12 小时前
自定义层和读写文件
pytorch·python·深度学习
似水流年 光阴已逝13 小时前
从Excel姓名匹配案例学Python:由点及面的系统化学习指南
开发语言·python·excel
这里有鱼汤13 小时前
量化的困局:当所有人都在跑同一个因子时,我们还能赚谁的钱?
后端·python
新子y13 小时前
【小白笔记】strip的含义
笔记·python
Q_Q196328847513 小时前
python+vue的在线租房 房屋租赁系统
开发语言·vue.js·spring boot·python·django·flask·node.js
空影星13 小时前
GridPlayer,一个好用的多视频同步播放器
python·flask·电脑·音视频
草莓熊Lotso13 小时前
《C++ Web 自动化测试实战:常用函数全解析与场景化应用指南》
前端·c++·python·dubbo
叼菠萝13 小时前
AI 应用开发三剑客系列:LangChain 如何撑起 LLM 应用开发基石?
python·langchain