完整树类

<?php
header('content-type:text/html;charset=utf-8');

class Category
{	
		//原始的分类数据
		private $rawList=array();
		//格式化后的分类
		private $formatList=array();
		//格式化的字符
		private $icon = array('│','├','└');
		//字段映射,分类id,上级分类pid,分类名称title,格式化后分类名称fulltitle
		private $field=array();
		
		/*
		功能:构造函数;
		属性:public;
		参数: $field,字段映射,分类id,上级分类pid,分类名称title,格式化后分类名称fulltitle
		依次传递,例如在分类数据表中,分类id,字段名为cid,上级分类pid,字段名称name,希望格式化分类后输出cname,
		  则,传递参数为,$field('cid','pid','name','cname');若为空,则采用默认值。
		*/		
		public function __construct($field=array())
		{			
				$this->field['id']=isset($field['0'])?$field['0']:'id';
				$this->field['pid']=isset($field['1'])?$field['1']:'pid';
				$this->field['title']=isset($field['2'])?$field['2']:'title';
				$this->field['fulltitle']=isset($field['3'])?$field['3']:'fulltitle';
		}

		/*
		功能:返回给定上级分类$pid的所有同一级子分类;
		属性:public;
		参数:上级分类$pid;
		返回:子分类,二维数组;
		*/		
		public function getChild($pid,$data=array())
		{
			$childs=array();
			if(empty($data))
			{
				$data=$this->rawList;
			}
			foreach($data as $Category)
			{
				if($Category[$this->field['pid']]==$pid)
					$childs[]=$Category;
			}
			return $childs;
		}
	
		/*
		功能:得到递归格式化分类;
		属性:public;
		参数:$data,二维数组,起始分类id,默认$id=0;
		返回:递归格式化分类数组;
		*/		
		public function getTree($data,$id=0)
		{
			//数据为空,则返回
			if(empty($data))
				return false;
				
			$this->rawList=array();
			$this->formatList=array();
			$this->rawList=$data;
			$this->_searchList($id);
			return $this->formatList;
		}
			

		private function _searchList($id=0,$space="")
		{
			//下级分类的数组
			$childs=$this->getChild($id);
			//如果没下级分类,结束递归
			if(!($n=count($childs)))
				return;
			$cnt=1;
			//循环所有的下级分类
			for($i=0;$i<$n;$i++)
			{
					$pre="";
					$pad="";
					if($n==$cnt)
					{
						$pre=$this->icon[2];
					}
					else
					{
						$pre=$this->icon[1];
					$pad=$space?$this->icon[0]:"";
				}
				$childs[$i][$this->field['fulltitle']]=($space?$space.$pre:"").$childs[$i][$this->field['title']];
				$this->formatList[]=$childs[$i];
				//递归下一级分类
				$this->_searchList($childs[$i][$this->field['id']],$space.$pad."  ");
				$cnt++;
			}
	}
}
	$data[]=array('cat_id'=>1,'pid'=>0,'name'=>'中国');
	$data[]=array('cat_id'=>2,'pid'=>0,'name'=>'美国');
	$data[]=array('cat_id'=>3,'pid'=>0,'name'=>'韩国');
	$data[]=array('cat_id'=>4,'pid'=>1,'name'=>'北京');
	$data[]=array('cat_id'=>5,'pid'=>1,'name'=>'上海');
	$data[]=array('cat_id'=>6,'pid'=>1,'name'=>'广西');
	$data[]=array('cat_id'=>7,'pid'=>6,'name'=>'桂林');
	$data[]=array('cat_id'=>8,'pid'=>6,'name'=>'南宁');
	$data[]=array('cat_id'=>9,'pid'=>6,'name'=>'柳州');
	$data[]=array('cat_id'=>10,'pid'=>2,'name'=>'纽约');
	$data[]=array('cat_id'=>11,'pid'=>2,'name'=>'华盛顿');
	$data[]=array('cat_id'=>12,'pid'=>3,'name'=>'首尔');
	
	$cat=new Category(array('cat_id','pid','name','cname'));
	$s=$cat->getTree($data);//获取分类数据树结构
	//$s=$cat->getTree($data,1);获取pid=1所有子类数据树结构
	foreach($s as $vo)
	{
	 echo $vo['cname'].'<br>';
	}
?>

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下是Python实现CART分类回归决策树的完整代码: ```python import numpy as np class CARTDecisionTree: def __init__(self): self.tree = {} def fit(self, X, y, feature_names=None, max_depth=5, min_samples_split=2): self.feature_names = feature_names self.max_depth = max_depth self.min_samples_split = min_samples_split self.tree = self.build_tree(X, y) def predict(self, X): return [self.predict_one(x, self.tree) for x in X] def predict_one(self, x, tree): if not isinstance(tree, dict): return tree feature, threshold, tree_dict = tree.values() if x[feature] <= threshold: return self.predict_one(x, tree_dict['left']) else: return self.predict_one(x, tree_dict['right']) def build_tree(self, X, y, depth=0): num_samples, num_features = X.shape num_labels = len(np.unique(y)) if depth == self.max_depth or num_labels == 1 or num_samples < self.min_samples_split: return self.get_leaf_node(y) best_feature, best_threshold = self.get_best_split(X, y, num_samples, num_features) left_indices = X[:, best_feature] <= best_threshold right_indices = X[:, best_feature] > best_threshold left_tree = self.build_tree(X[left_indices], y[left_indices], depth + 1) right_tree = self.build_tree(X[right_indices], y[right_indices], depth + 1) return {'feature': best_feature, 'threshold': best_threshold, 'left': left_tree, 'right': right_tree} def get_best_split(self, X, y, num_samples, num_features): best_feature = None best_threshold = None best_gini = 1 for feature in range(num_features): thresholds, classes = zip(*sorted(zip(X[:, feature], y))) num_left_samples = 0 num_left_labels = {} num_right_samples = num_samples num_right_labels = {} for i in range(1, num_samples): label = classes[i-1] num_left_samples += 1 num_left_labels[label] = num_left_labels.get(label, 0) + 1 num_right_samples -= 1 num_right_labels[label] = num_right_labels.get(label, 0) + 1 if thresholds[i] == thresholds[i-1]: continue left_gini = self.get_gini(num_left_labels, num_left_samples) right_gini = self.get_gini(num_right_labels, num_right_samples) gini = (num_left_samples * left_gini + num_right_samples * right_gini) / num_samples if gini < best_gini: best_gini = gini best_feature = feature best_threshold = (thresholds[i] + thresholds[i-1]) / 2 return best_feature, best_threshold def get_gini(self, label_counts, num_samples): impurity = 1 for label in label_counts: prob = label_counts[label] / num_samples impurity -= prob ** 2 return impurity def get_leaf_node(self, y): label_counts = {} for label in y: label_counts[label] = label_counts.get(label, 0) + 1 return max(label_counts, key=label_counts.get) ``` 其中,`fit` 方法用于拟合训练数据,`predict` 方法用于预测测试数据。我们还实现了 `build_tree` 方法用于构建决策树,`predict_one` 方法用于对单个样本进行预测,`get_best_split` 方法用于找到最佳的分裂点,`get_gini` 方法用于计算基尼不纯度,`get_leaf_node` 方法用于生成叶子节点。 在使用时,我们可以先创建一个决策树对象,然后调用其 `fit` 方法进行训练,最后调用 `predict` 方法进行预测,如下所示: ```python X_train = np.array([[0, 0], [0, 1], [1, 0], [1, 1]]) y_train = np.array([0, 1, 1, 0]) X_test = np.array([[0, 0], [0, 1], [1, 0], [1, 1]]) dt = CARTDecisionTree() dt.fit(X_train, y_train) y_pred = dt.predict(X_test) print(y_pred) ``` 输出结果为:`[0, 1, 1, 0]`,与预期结果相符。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值