I’m new to python. Some help would be great, thanks in advance too. I am trying to get the various average salary based on Units.
In Excel, I would do =AVERAGE(FILTER(B:B,A:A=D3)). Where Column B is all the individual salaries and Column A is the various Unit.
I have managed to do an array of the different units:
DiffUnits = df["Unit"].unique()
My data:
| Units | Salary |
|---|---|
| IT | 500 |
| Math | 3000 |
| Math | 1200 |
| Science | 700 |
| IT | 2000 |
| Science | 1800 |
Expected result (In the form of a table, if possible):
| Units | AverageSalary |
|---|---|
| IT | 1250 |
| Math | 2100 |
| Science | 1250 |
Advertisement
Answer
import pandas as pd
df = pd.DataFrame({'Units': ['IT', 'Math', 'Math', 'Science', 'IT', 'Science'],
'Salary': [500, 3000, 1200, 700, 2000, 1800]})
df2 = df.groupby('Units').mean().reset_index()
df2
gives you:
Units Salary 0 IT 1250 1 Math 2100 2 Science 1250