Excel VBA to Python code (Filter function)

Question

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

In Excel, you could have used AVERAGEIF/S.

Parfait
– Parfait

2022-07-30 17:08:13 +00:00
Commented Jul 30, 2022 at 17:08 — Parfait
– Parfait, Commented Jul 30, 2022 at 17:08

John M. · Accepted Answer · 2022-07-30 15:39:53Z

2

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

answered Jul 30, 2022 at 15:39

John M.

8056 silver badges16 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Jayl Over a year ago

Thank you! Didn't know 1 line would work already...

Parfait Over a year ago

Use as_index=False argument of groupby and avoid reset_index.

Collectives™ on Stack Overflow

Excel VBA to Python code (Filter function)

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related