Python Forum
Create new column in dataframe
Thread Rating:
  • 0 Vote(s) - 0 Average
  • 1
  • 2
  • 3
  • 4
  • 5
Create new column in dataframe
#10
import pandas as pd

data = {
    'Date': ['1/12/2002', '1/12/2002', '1/12/2002', '1/12/2002', '1/01/2003', '1/01/2003', '1/01/2003', '1/01/2003', '1/01/2003'],
    'id': [1, 1, 1, 2, 1, 1, 2, 2, 2],
    'class': ['L', 'H', 'L', 'L', 'L', 'H', 'H', 'L', 'L'],
    'amount': [5, 6, 7, 3, 2, 8, 7, 9, 1]
}

df = pd.DataFrame(data)
# Pivot the DataFrame
pivot_df = df.pivot_table(index=['Date', 'id'], columns='class', values='amount', aggfunc='sum', fill_value=0)
pivot_df.columns = [f'{col}_class' for col in pivot_df.columns]
# Reset index to get Date and id back as columns
pivot_df = pivot_df.reset_index()
print(pivot_df)
Output:
Date id H_class L_class 0 1/01/2003 1 8 2 1 1/01/2003 2 7 10 2 1/12/2002 1 6 12 3 1/12/2002 2 0 3
Pedroski55 likes this post
Reply


Messages In This Thread
Create new column in dataframe - by Scott - Jun-14-2024, 07:04 AM
RE: Create new column in dataframe - by Larz60+ - Jun-14-2024, 07:23 AM
RE: Create new column in dataframe - by Scott - Jun-16-2024, 10:40 PM
RE: Create new column in dataframe - by Larz60+ - Jun-16-2024, 11:47 PM
RE: Create new column in dataframe - by Scott - Jun-21-2024, 05:09 AM
RE: Create new column in dataframe - by Scott - Jun-21-2024, 05:20 AM
RE: Create new column in dataframe - by Larz60+ - Jun-21-2024, 08:03 PM
RE: Create new column in dataframe - by deanhystad - Jun-21-2024, 10:36 PM
RE: Create new column in dataframe - by Pedroski55 - Jun-23-2024, 07:27 AM
RE: Create new column in dataframe - by snippsat - Jun-23-2024, 08:52 PM

Possibly Related Threads…
Thread Author Replies Views Last Post
  attempt to split values from within a dataframe column mbrown009 9 2,782 Jun-20-2024, 07:59 PM
Last Post: AdamHensley
  Putting column name to dataframe, can't work. jonah88888 2 2,052 Jun-18-2024, 09:19 PM
Last Post: AdamHensley
  concat 3 columns of dataframe to one column flash77 2 1,064 Oct-03-2023, 09:29 PM
Last Post: flash77
  HTML Decoder pandas dataframe column mbrown009 3 1,283 Sep-29-2023, 05:56 PM
Last Post: deanhystad
  New Dataframe Column Based on Several Conditions nb1214 1 1,933 Nov-16-2021, 10:52 PM
Last Post: jefsummers
  Setting the x-axis to a specific column in a dataframe devansing 0 2,152 May-23-2021, 12:11 AM
Last Post: devansing
Question [Solved] How to refer to dataframe column name based on a list lorensa74 1 2,436 May-17-2021, 07:02 AM
Last Post: lorensa74
Question Pandas - Creating additional column in dataframe from another column Azureaus 2 3,132 Jan-11-2021, 09:53 PM
Last Post: Azureaus
  Filter data based on a value from another dataframe column and create a file using lo pawanmtm 1 4,422 Jul-15-2020, 06:20 PM
Last Post: pawanmtm
  Pandas DataFrame and unmatched column sritsv19 0 3,155 Jul-07-2020, 12:52 PM
Last Post: sritsv19

Forum Jump:

User Panel Messages

Announcements
Announcement #1 8/1/2020
Announcement #2 8/2/2020
Announcement #3 8/6/2020