python的等深分箱实例
作者:Damon_guang 时间:2022-11-30 01:01:20
背景
当前很多文章尝试过最优分箱,python上也有cut等方法进行等宽分箱。为了方便日后输出结果以及分箱要求。做一个简单的轮子以供大家日后使用。很多能用其他轮子的地方也没有多余出力,也不托大会比别人写的好。空间复杂度尽我所能。
方法展示
话不多说上代码。
以下为等深分箱以及encoding方法
# -*- coding: utf-8 -*-
"""
Created on Tue Jan 29 17:26:38 2019
@author: DamomWCG
"""
class Equal_depth_box:
def equal_box(list, bin_num):
'''
param:
list:you need bin box list
bin_num: you want bin num
'''
bin_num = 10
list.sort() #need sort can replace by others
list_2 = list.copy()
all_num = len(list_2)
bin_sep = all_num/bin_num
bin_sep = int(bin_sep)
bin_list = []
for i in range(1,bin_num):
bin_dict = {}
bin_dict = i*bin_sep
bin_list.append(bin_dict)
bin_real_list = []
for i in bin_list:
#print(i)
bin_real_dict = {}
bin_real_dict = list[i]
bin_real_list.append(bin_real_dict)
return bin_real_list
def replace_box(list_1,replace_list):
'''
param:
list_1:you need bin list
replace_list: from equal box, replace the original list
'''
import pandas as pd
list_max = max(list_1)
list_min = min(list_1)
replace_list.insert(0,list_min -1) #insert start
replace_list.append(list_max + 1) #insert end
list_2 = pd.cut(list_1, bins = replace_list,
labels = range(len(replace_list) - 1)).tolist()
return list_2
if __name__ == '__main__':
import random
list_1 = random.sample(range(1000), 134)
print(list_1.sort())
print('real_list: {}'.format(list_1[0:50]))
replace_list = bin_class.equal_box(list_1, 10)
list_2 = bin_class.replace_box(list_1, replace_list)
print('encode_list: {}'.format(list_2[0:50]))
原始git地址
https://github.com/DamonWCG/Equal_depth_box/blob/master/Equal_depth_box
使用案例
本案例数据为,我进行过一步的最优分箱,来做一步等深分箱来进行横向对比。
原始数据形式
具体案例
# -*- coding: utf-8 -*-
"""
Created on Wed Jan 30 16:17:19 2019
@author: DamonWCG
"""
from Equal_depth_box import *
import pandas as pd
df = pd.read_csv('test.csv', encoding = 'gbk')
df.columns
####需要分箱的列
list_1 = df['deal_city_encoding'].tolist() #本方法是针对于list,所以对于series需要进行变换
####需要分箱的个数
replace_list = Equal_depth_box.equal_box(list_1, 5)
####替代的名称
##因为列表排序所以需要重新排序对齐,这里我有空再想想其他办法
df.sort_values(by="deal_city_encoding", inplace = True)
list_2 = Equal_depth_box.replace_box(list_1, replace_list)
df['deal_city_bin_encoding'] = list_2
df.to_csv('df.csv', encoding = 'gbk', index = False)
本方法现阶段encoding形式为数字型。
分箱最终结果形式
需要注意,我得列表形式重新decode时候需要按照当前排序数组重新编码,这个问题我年后会解决。
来源:https://blog.csdn.net/weixin_40192436/article/details/86706231
标签:python,等深分箱
0
投稿
猜你喜欢
python简单商城购物车实例代码
2022-10-28 02:16:37
删除目录下相同文件的python代码(逐级优化)
2023-08-02 17:25:10
Python seek()和tell()函数的具体使用
2023-12-07 20:59:39
条件注释使用指南[译]
2009-03-23 17:41:00
基于vue v-for 循环复选框-默认勾选第一个的实现方法
2024-05-13 09:38:01
SQL Server Agent 服务启动后又停止问题
2024-01-25 21:56:49
Python计算回文数的方法
2022-04-02 05:08:07
解决Django中checkbox复选框的传值问题
2023-09-11 17:15:22
Python+Tableau广东省人口普查可视化的实现
2022-02-12 21:53:59
详解Python字符串切片
2021-09-10 05:10:43
python 获取图片分辨率的方法
2022-02-19 12:03:56
python中异常的传播详解
2021-12-20 11:31:05
python之PyAutoGui教你做个自动脚本计算器的方法
2022-01-08 15:24:55
使用Python的toolz库开始函数式编程的方法
2022-10-18 10:29:55
Vue.js -- 过滤器使用总结
2024-05-09 10:41:28
python 通过视频url获取视频的宽高方式
2022-06-04 08:10:58
如何查询Top N及Top(M―N)记录?
2009-11-11 20:03:00
vue使用swiper实现中间大两边小的轮播图效果
2024-06-05 09:20:25
mysql中如何查询数据库中的表名
2024-01-20 03:07:32
python 实现简单的计算器(gui界面)
2022-11-14 14:35:14