【Python】小技巧——参数的可视化

参考链接：https://mp.weixin.qq.com/s/kztRjHYJRJzG0h9pjfdwBw，公众号：Crossin的编程教室

看到一个小技巧觉得很有用，记录学习一下。

如果通过模型生成了参数，但是系数小数位多且个数多，可以通过style.bar可视化的展示一下。

 

一、示例数据如下：

1 import numpy as np
2 import pandas as pd
3 
4 coef = np.array([-0.01473407, -0.14209676, 0.0033682, 0.30169562, -0.89432301, -0.38844689, 0.0731774, .39652068])
5 
6 variables = np.array(['Age','SibSp','Fare','Parch','Pclass','Embarked_S','Embarked_C','Embarked_Q'])
7  
8 pd.DataFrame({'variable':variables,'coefficient':coef})

 

运行的结果如下：

variable    coefficient
0    Age    -0.014734
1    SibSp    -0.142097
2    Fare    0.003368
3    Parch    0.301696
4    Pclass    -0.894323
5    Embarked_S    -0.388447
6    Embarked_C    0.073177
7    Embarked_Q    0.396521

 

二、参数可视化：

data = pd.DataFrame({'variable':variables,'coefficient':coef})
data.round(decimals=2).sort_values('coefficient',ascending=False).style.bar(color=['grey','lightblue'],align='zero')

效果如下（保留两位小数→降序排列→可视化）：

 

 

注：`align` must be one of {'left', 'zero',' mid'}——参数还需要仔细体会下