决策树代码python

以下是一个简单的决策树代码示例，使用sklearn库：

from sklearn import tree

# 准备数据
features = [[140, 1], [130, 1], [150, 0], [170, 0]]  # 特征：重量（克），表面光滑（1）或粗糙（0）
labels = [0, 0, 1, 1]  # 标签：苹果（0）或橙子（1）

# 训练模型
clf = tree.DecisionTreeClassifier()
clf = clf.fit(features, labels)

# 预测
print(clf.predict([[160, 0]]))  # 预测一枚重160克、表面粗糙的水果是什么

输出：

[1]

这表示这枚水果被预测为橙子。