用Python将鸢尾花数据集中的一个属性转化为多元伯努利变量并给出代码
时间: 2024-05-14 11:13:08 浏览: 94
机器学习鸢尾花数据集
以下是将鸢尾花数据集中的一个属性转化为多元伯努利变量的Python代码:
```python
import pandas as pd
# 加载鸢尾花数据集
iris = pd.read_csv('iris.csv')
# 将'petal_width'属性转化为多元伯努利变量
iris['petal_width_bin'] = pd.cut(iris['petal_width'], bins=[0, 0.75, 1.5, 2.0], labels=[0, 1, 2])
# 显示前5行数据
print(iris.head())
```
这段代码将'petal_width'属性分成三个区间,然后用0、1、2三个数字分别表示每个区间,从而将该属性转化为多元伯努利变量。
阅读全文