蘑菇数据集中特征与有毒属性相关性分析及可视化
该代码主要针对蘑菇数据集中各特征与有毒属性之间的相关性进行分析,并通过可视化方式呈现。
第94行代码定义了一个名为'analysis_poison'的函数,用于计算各特征与有毒属性的相关性。该函数传入两个参数,一个是数据集(data),另一个是特征名称(index_name)。在函数内部,首先将数据集中的有毒属性('class')进行替换,将其转换为数字类型(有毒为1,无毒为0)。然后,通过groupby()函数对指定的特征进行分组,计算有毒属性的和,再除以该特征的总数量,得到该特征与有毒属性的相关性。
第98-111行代码用于可视化各特征与有毒属性的相关性。首先,通过plt.close()函数关闭之前的图像。然后,定义一个16行30列的画布,并初始化计数器i为1。接着,通过for循环遍历数据集中的每一个特征(除了'class'属性),调用analysis_poison()函数计算其与有毒属性的相关性,并在画布的第i个位置上绘制柱状图。同时,将相关性大于0.75的特征存入列表danger中,最后将计数器i加1。最后通过plt.show()函数展示图像。
原文地址: https://www.cveoy.top/t/topic/owC9 著作权归作者所有。请勿转载和采集!