task.py

import pandas
import numpy
import matplotlib.pyplot as plt

works = pandas.read_csv("works.csv")
print(works['skills'].str.lower().str.contains('python|питон'))

#task1
works = pandas.read_csv("works.csv")
head = works.head(5)
print(head)

tail = works.tail(5)
print(tail)
print(works.shape[0])
print(len(works.index))

#task2
print(works[works['gender'] == 'Мужской'].shape[0])
print((works['gender'] == 'Женский').sum())
print(works['gender'].value_counts())

#task3
print(works['skills'].notnull().sum())
print(works.info())
print(works['skills'].count())

#task4
print(works[works['skills'].notnull()]['skills'])
print(works['skills'].dropna())
print(works.query("skills == skills")["skills"])
print(works.query("salary == 15000"))
edu = 'Высшее'
gen = 'Женский'
print(works.query("educationType == @edu and gender == @gen")[['salary', 'educationType','gender']])

#task5
mask = works["skills"].str.lower().str.contains("python|питон") & works["skills"].notnull()
print(works[mask]["salary"])

#task6
percentiles = numpy.linspace(.1, 1, 10)

gen = "Мужской"
men_salary = works.query('gender == @gen').quantile(percentiles)
fig, ax = plt.subplots()
ax.plot(percentiles, men_salary)
plt.xlabel('Перцентили')
plt.ylabel('Зарплата мужчин')
plt.show()

gen = "Женский"
women_salary = works.query('gender == @gen').quantile(percentiles)
fig, ax = plt.subplots()
ax.plot(percentiles, women_salary)
plt.xlabel('Перцентили')
plt.ylabel('Зарплата женщин')
plt.show()

#task7
gen = "Мужской"
men_salary = works.query('gender == @gen').groupby("educationType").agg("mean").reset_index()
men = men_salary['salary'].values
gen = "Женский"
women_salary = works.query('gender == @gen').groupby("educationType").agg("mean").reset_index()
women = women_salary['salary'].values

types = men_salary["educationType"].values
id = numpy.arange(len(types))

plt.bar(id - 0.2, men, 0.4, color="g", label = "Средняя зарплата мужчин")
plt.bar(id + 0.2, women, 0.4, color="y", label = "Средняя зарплата женщин")
plt.xticks(id, types, rotation=45)
plt.legend()
plt.show()