侯体宗的博客
  • 首页
  • Hyperf版
  • beego仿版
  • 人生(杂谈)
  • 技术
  • 关于我
  • 更多分类
    • 文件下载
    • 文字修仙
    • 中国象棋ai
    • 群聊
    • 九宫格抽奖
    • 拼图
    • 消消乐
    • 相册

Python绘制KS曲线的实现方法

Python  /  管理员 发布于 7年前   151

python实现KS曲线,相关使用方法请参考上篇博客-R语言实现KS曲线

代码如下:

####################### PlotKS ##########################def PlotKS(preds, labels, n, asc):    # preds is score: asc=1  # preds is prob: asc=0    pred = preds # 预测值  bad = labels # 取1为bad, 0为good  ksds = DataFrame({'bad': bad, 'pred': pred})  ksds['good'] = 1 - ksds.bad    if asc == 1:    ksds1 = ksds.sort_values(by=['pred', 'bad'], ascending=[True, True])  elif asc == 0:    ksds1 = ksds.sort_values(by=['pred', 'bad'], ascending=[False, True])  ksds1.index = range(len(ksds1.pred))  ksds1['cumsum_good1'] = 1.0*ksds1.good.cumsum()/sum(ksds1.good)  ksds1['cumsum_bad1'] = 1.0*ksds1.bad.cumsum()/sum(ksds1.bad)    if asc == 1:    ksds2 = ksds.sort_values(by=['pred', 'bad'], ascending=[True, False])  elif asc == 0:    ksds2 = ksds.sort_values(by=['pred', 'bad'], ascending=[False, False])  ksds2.index = range(len(ksds2.pred))  ksds2['cumsum_good2'] = 1.0*ksds2.good.cumsum()/sum(ksds2.good)  ksds2['cumsum_bad2'] = 1.0*ksds2.bad.cumsum()/sum(ksds2.bad)    # ksds1 ksds2 -> average  ksds = ksds1[['cumsum_good1', 'cumsum_bad1']]  ksds['cumsum_good2'] = ksds2['cumsum_good2']  ksds['cumsum_bad2'] = ksds2['cumsum_bad2']  ksds['cumsum_good'] = (ksds['cumsum_good1'] + ksds['cumsum_good2'])/2  ksds['cumsum_bad'] = (ksds['cumsum_bad1'] + ksds['cumsum_bad2'])/2    # ks  ksds['ks'] = ksds['cumsum_bad'] - ksds['cumsum_good']  ksds['tile0'] = range(1, len(ksds.ks) + 1)  ksds['tile'] = 1.0*ksds['tile0']/len(ksds['tile0'])    qe = list(np.arange(0, 1, 1.0/n))  qe.append(1)  qe = qe[1:]    ks_index = Series(ksds.index)  ks_index = ks_index.quantile(q = qe)  ks_index = np.ceil(ks_index).astype(int)  ks_index = list(ks_index)    ksds = ksds.loc[ks_index]  ksds = ksds[['tile', 'cumsum_good', 'cumsum_bad', 'ks']]  ksds0 = np.array([[0, 0, 0, 0]])  ksds = np.concatenate([ksds0, ksds], axis=0)  ksds = DataFrame(ksds, columns=['tile', 'cumsum_good', 'cumsum_bad', 'ks'])    ks_value = ksds.ks.max()  ks_pop = ksds.tile[ksds.ks.idxmax()]  print ('ks_value is ' + str(np.round(ks_value, 4)) + ' at pop = ' + str(np.round(ks_pop, 4)))    # chart  plt.plot(ksds.tile, ksds.cumsum_good, label='cum_good', color='blue', linestyle='-', linewidth=2)   plt.plot(ksds.tile, ksds.cumsum_bad, label='cum_bad',color='red', linestyle='-', linewidth=2)  plt.plot(ksds.tile, ksds.ks, label='ks',          color='green', linestyle='-', linewidth=2)  plt.axvline(ks_pop, color='gray', linestyle='--')  plt.axhline(ks_value, color='green', linestyle='--')  plt.axhline(ksds.loc[ksds.ks.idxmax(), 'cumsum_good'], color='blue', linestyle='--')  plt.axhline(ksds.loc[ksds.ks.idxmax(),'cumsum_bad'], color='red', linestyle='--')  plt.title('KS=%s ' %np.round(ks_value, 4) +         'at Pop=%s' %np.round(ks_pop, 4), fontsize=15)    return ksds####################### over ##########################

作图效果如下:

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。


  • 上一条:
    python 实现A*算法的示例代码
    下一条:
    Python标准库shutil用法实例详解
  • 昵称:

    邮箱:

    0条评论 (评论内容有缓存机制,请悉知!)
    最新最热
    • 分类目录
    • 人生(杂谈)
    • 技术
    • linux
    • Java
    • php
    • 框架(架构)
    • 前端
    • ThinkPHP
    • 数据库
    • 微信(小程序)
    • Laravel
    • Redis
    • Docker
    • Go
    • swoole
    • Windows
    • Python
    • 苹果(mac/ios)
    • 相关文章
    • 在python语言中Flask框架的学习及简单功能示例(0个评论)
    • 在Python语言中实现GUI全屏倒计时代码示例(0个评论)
    • Python + zipfile库实现zip文件解压自动化脚本示例(0个评论)
    • python爬虫BeautifulSoup快速抓取网站图片(1个评论)
    • vscode 配置 python3开发环境的方法(0个评论)
    • 近期文章
    • 在go中实现一个常用的先进先出的缓存淘汰算法示例代码(0个评论)
    • 在go+gin中使用"github.com/skip2/go-qrcode"实现url转二维码功能(0个评论)
    • 在go语言中使用api.geonames.org接口实现根据国际邮政编码获取地址信息功能(1个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf分页文件功能(0个评论)
    • gmail发邮件报错:534 5.7.9 Application-specific password required...解决方案(0个评论)
    • 欧盟关于强迫劳动的规定的官方举报渠道及官方举报网站(0个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf文件功能(0个评论)
    • Laravel从Accel获得5700万美元A轮融资(0个评论)
    • 在go + gin中gorm实现指定搜索/区间搜索分页列表功能接口实例(0个评论)
    • 在go语言中实现IP/CIDR的ip和netmask互转及IP段形式互转及ip是否存在IP/CIDR(0个评论)
    • 近期评论
    • 122 在

      学历:一种延缓就业设计,生活需求下的权衡之选中评论 工作几年后,报名考研了,到现在还没认真学习备考,迷茫中。作为一名北漂互联网打工人..
    • 123 在

      Clash for Windows作者删库跑路了,github已404中评论 按理说只要你在国内,所有的流量进出都在监控范围内,不管你怎么隐藏也没用,想搞你分..
    • 原梓番博客 在

      在Laravel框架中使用模型Model分表最简单的方法中评论 好久好久都没看友情链接申请了,今天刚看,已经添加。..
    • 博主 在

      佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了,可以看看近期评论的其他文章..
    • 1111 在

      佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开,博主百忙中能否发个APP下载链接,佛跳墙或极光..
    • 2016-10
    • 2016-11
    • 2018-04
    • 2020-03
    • 2020-04
    • 2020-05
    • 2020-06
    • 2022-01
    • 2023-07
    • 2023-10
    Top

    Copyright·© 2019 侯体宗版权所有· 粤ICP备20027696号 PHP交流群

    侯体宗的博客