代码:
import pandas as pd
import numpy as np
sample_set = pd.read_excel(‘C:\Users\四维空间里的散点\Desktop\190225.xls’)
time = sample_set[‘经营年限’]
def calc_score_median(sample_set, var):
‘’’
计算相邻评分的中位数,以便进行决策树二元切分
param sample_set: 待切分样本
param var: 分割变量名称
'''
var_list = list(np.unique(sample_set[var]))
var_median_list = []
for i in range(len(var_list) - 1):
var_median = (var_list[i] + var_list[i + 1]) / 2
var_median_list.append(var_median)
return var_median_list
time_list = calc_score_median(sample_set,time)
报错:
File “C:\Users\四维空间里的散点\AppData\Roaming\Python\Python37\site-packages\pandas\core\indexing.py”, line 1354, in _convert_to_indexer
return self._get_listlike_indexer(obj, axis, **kwargs)[1]
File “C:\Users\四维空间里的散点\AppData\Roaming\Python\Python37\site-packages\pandas\core\indexing.py”, line 116