Как создать функцию для визуального распределения заработной платы сотрудников? - PullRequest
0 голосов
/ 25 октября 2019

Я хотел бы видеть распределение заработной платы по категориям работы сотрудников, чтобы я мог видеть, переплачивают ли они кому-то или недоплачивают.

Вот часть моего набора данных:

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

Статус означает, уволен ли сотрудник или нет. 1 означает нет, 0 означает да.

Итак, сначала я выбрал все еще активных сотрудников в качестве активных = data [data ['Status'] == 1]

Но я понятия не имею, как выбрать категорию вакансии, чтобысоздать соответствующий график распределения для распределения заработной платы.

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

Как добавить столбец Зарплата в коде?

Есть ли способ определить функцию, которая каждый раз, когда я ввожу категорию, тогда даст мне график для зарплатыраспределение ??

Я новичок в алгоритмах и хотел бы знать, если кто-нибудь может помочь.

1 Ответ

1 голос
/ 25 октября 2019

Вы можете определить функцию следующим образом

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

plotSalary('Analyst')
...