/Python/Add a new categorical column to an existing table python

Add a new categorical column to an existing table python

d = df.groupby('Item_Identifier')['Sales'].mean().to_dict()
print (d)
{'Beef': 3030.0, 'Milk': 1233.3333333333333, 'Tea': 150.0}

print (df['Item_Identifier'].map(d))
0    1233.333333
1    1233.333333
2    1233.333333
3    3030.000000
4    3030.000000
5     150.000000
6     150.000000
7     150.000000
Name: Item_Identifier, dtype: float64

bins = [df['Sales'].min(),500, 1500, df['Sales].max()]
labels=['low','medium','high']
df['Price Category'] = pd.cut(df['Item_Identifier'].map(d), bins=bins, labels=labels)
print (df)
  Item_Identifier  Sales Price Category
0            Milk    500         medium
1            Milk   1200         medium
2            Milk   2000         medium
3            Beef     60           high
4            Beef   6000           high
5             Tea    150            low
6             Tea    100            low
7             Tea    200            low

Source: stackoverflow.com

Similar codes

-1 vote107 views1 answer
How to call a class in python
Python
0 vote151 views1 answer
Print the multiple data types in a single program in list data structures
Python
1 vote120 views1 answer
Values outside range pandas
Python
0 vote127 views1 answer
A GDAL API version must be specified. Provide a path to gdal-config using a GDAL_CONFIG environment variable or use a GDAL_VERSION environment variable.
Python
2 votes177 views1 answer
Append row to array python
Python
0 vote112 views1 answer
Print a commans in python
Python
0 vote154 views1 answer
Pip install contractions
Python
1 vote97 views3 answers
Python create file if not exists
Python
22 votes117 views5 answers
If else python
Python
0 vote172 views3 answers
How to create a User and User profile in django rest framework
Python