python - 如何创建可视化员工工资分布的函数？

Question

我想按员工的工作类别查看工资分布，这样我就可以看到有人是多付还是少付。

这是我的数据集的一部分：

Category        Salary      Status
Analyst          35           1
Analyst          42           0
Manager          55           0
Administrator    25           1
..................................

状态是指员工是否已终止。1 表示否，0 表示是。

所以，首先我选择仍然活跃的员工为 active = data[data['Status']==1]

但我不知道如何选择工作类别以便为工资分布创建相应的分布图。

sns.distplot(active.Category =='Analyst',kde=False, bins=30)

如何在代码中添加 Salary 列？

有没有办法定义一个函数，每次我输入类别时它都会给我工资分布图？

我是算法的新手，想知道是否有人可以提供帮助。

标签： pythondata-visualizationseaborn

Answer 1

您可以如下定义函数

def plotSalary(category):
    sns.distplot(active[active.Category ==category]['Salary'],kde=False, bins=30)

plotSalary('Analyst')