Pandas Pivot Confusion

Asked Jan 20 '23 at 18:29

Active Jan 20 '23 at 19:31

Viewed 26 times

Problem

Converting R to Python via Pandas, and I am trying to using pd.DataFrame.pivot to accomplish the following.

I am trying to turn the following table:

type	life	year_1	year_2	year_3	year_4	year_5	cost
A	4	0	22	24	24	315	0
B	1	35	35	35	35	35	1

into this other table

type	life	cost	years	units
A	4	0	year_one	0
A	4	0	year_two	22
A	4	0	year_three	24
A	4	0	year_four	24
A	4	0	year_five	315
B	1	1	year_one	35
B	1	1	year_two	35
B	1	1	year_three	35
B	1	1	year_four	35
B	1	1	year_five	35

The R code I am converting is the following:

datasheet %>% pivot_longer(cols = starts_with("year_"),
                           names_to = "year", 
                           values_to = "units")

edited Jan 20 '23 at 19:31

M--

25,431
8
61
93

asked Jan 20 '23 at 18:29

James Adams

1

Use from melt method. Read about it in pandas doc – Alireza75 Jan 20 '23 at 18:31
1

Or you can use from wide_to_long method – Alireza75 Jan 20 '23 at 18:31
@JilberUrbina I think you may have ran the R code, which should work. I am trying to find a python/pandas equivalent. – James Adams Jan 20 '23 at 18:37
2

melt should work : `df.melt(['type','life','cost'],var_name='years',value_name='units')` – anky Jan 20 '23 at 18:38

Pandas Pivot Confusion

Problem

0 Answers0