使用python绘制水质数据.csv的决策树

好的，您可以按照以下步骤来使用Python绘制csv文件中水质数据的决策树：

1. 导入必要的库

import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn import tree
import matplotlib.pyplot as plt

2. 读取CSV文件数据

data = pd.read_csv('水质数据.csv')

3. 定义特征和目标列

# 特征列
features = ['PH','溶解氧','电导率','浊度','氨氮']
# 目标列
target = '水质等级'

4. 设置训练数据

train_data = data[features]
train_label = data[target]

5. 构建决策树模型

model = DecisionTreeClassifier(criterion='entropy')
model.fit(train_data, train_label)

6. 绘制决策树图形

plt.figure(dpi=300)
tree.plot_tree(model,
               filled=True,
               feature_names=features,
               class_names=['I类', 'II类', 'III类', 'IV类', 'V类'],
               )
plt.show()

以上是使用Python绘制csv文件中水质数据的决策树的方法，希望可以帮到您。