Pandas 1.1.4
MRE:
JavaScript
x
10
10
1
df = pd.DataFrame({"Code":[1,2], "view_A":[3000, 2300], "click_A":[3, 23],
2
"view_B":[1200, 300], "click_B":[5, 3]})
3
df.set_index("Code", inplace=True)
4
5
>>>
6
view_A click_A view_B click_B
7
Code
8
1 3000 3 1200 5
9
2 2300 23 300 3
10
Want to make it into
JavaScript
1
7
1
view click
2
Code type
3
1 A 3000 3
4
2 A 2300 23
5
1 B 1200 5
6
2 B 300 3
7
I can do it, but want to explore more (clean) options.
My sol’tn
JavaScript
1
7
1
a_df = df[["view_A", "click_A"]].rename(columns={"view_A":"view", "click_A":"click"})
2
a_df["type"] = "A"
3
4
b_df = df[["view_B", "click_B"]].rename(columns={"view_B":"view", "click_B":"click"})
5
b_df["type"] = "B"
6
final_df = pd.concat([a_df, b_df])
7
But code is dirty.
Advertisement
Answer
This is essentially a reshape operation using stack
JavaScript
1
3
1
df.columns = df.columns.str.split('_', expand=True)
2
df.stack().rename_axis(['code', 'type'])
3
JavaScript
1
7
1
click view
2
code type
3
1 A 3 3000
4
B 5 1200
5
2 A 23 2300
6
B 3 300
7