帮我用pd.get_dummies写中风预测数据集处理分类变量的代码

好的，以下是使用pd.get_dummies函数对中风预测数据集进行分类变量处理的代码：

import pandas as pd

# 读取中风预测数据集
data = pd.read_csv('stroke_prediction.csv')

# 对分类变量进行处理
data = pd.get_dummies(data, columns=['gender', 'ever_married', 'work_type', 'Residence_type', 'smoking_status'], drop_first=True)

# 查看处理后的数据集
print(data.head())

上述代码中，`pd.get_dummies()`函数用于将指定的列进行独热编码处理，其中`columns`参数指定需要进行处理的列名，`drop_first=True`表示删除第一个编码列以避免多重共线性问题。最后将处理后的数据集打印输出，以便查看处理结果。