Grouping and pivoting dataframe

Question

I have a dataframe that looks like this:

Names         Company    Values   Period
HeadCount     Google     1000     1
HoursWorked   Google     500      1
HeadCount     Microsoft  600      1
HoursWorked   Microsoft  200      1
HeadCount     Google     2000     2
HoursWorked   Google     100      2

I'd like to transform this dataframe to something like this:

Company   Period  HeadCount  HoursWorked
Google    1       1000       500
Google    2       2000       100
Microsoft 1       600        200

Is there a way to take the columns pivot them, but also group by the company period and have a dataframe that looks like that?

Corralien · Accepted Answer · 2021-09-21 20:25:24Z

2

Use pivot_table:

>>> df.pivot_table(index=['Company', 'Period'], columns='Names', values='Values') \
      .rename_axis(None, axis=1).reset_index()

     Company  Period  HeadCount  HoursWorked
0     Google       1       1000          500
1     Google       2       2000          100
2  Microsoft       1        600          200

Edit:

I'm getting this error: all arrays must be same length

Solution: replace pivot by pivot_table.

edited Sep 21, 2021 at 20:25

answered Sep 21, 2021 at 19:54

Corralien

121k8 gold badges43 silver badges68 bronze badges

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

5 Comments

user2896120 Over a year ago

I'm getting this error: all arrays must be same length

Corralien Over a year ago

Is it possible to create a sample with the error?

Corralien Over a year ago

Try to do just df.pivot. Is the error still there?

user2896120 Over a year ago

I just changed it to pivot_table and it worked

Corralien Over a year ago

Strange but ok. I did not manage to reproduce the error. I updated my answer (if you want to accept it)

Collectives™ on Stack Overflow

Grouping and pivoting dataframe

1 Answer 1

5 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

5 Comments

Your Answer

Sign up or log in

Post as a guest

Related