Categorical Data Types in Pandas

Creating categorical data types

import pandas as pd
my_dict={'grade':['a', 'c', 'b', 'b','b','c']} # dictionary 
my_data = pd.DataFrame(data=my_dict)
my_data['grade']=my_data['grade'].astype('category')
print(my_data.dtypes)

grade    category
dtype: object

Using series.

import pandas as pd
my_df=pd.Series(['d','a','b','c','a','c'],dtype='category')
print(my_df.dtypes) # category

Using cut()

import pandas as pd 
my_dict={'NAME':['Ravi','Raju','Alex','Ron','King','Jack'],
         'ID':[1,2,3,4,5,6],
         'MATH':[80,40,70,70,60,30],
         'ENGLISH':[80,70,40,50,60,30]}
my_data = pd.DataFrame(data=my_dict)
my_data['my_cut'] = pd.cut(x=my_data['MATH'],bins=[1, 50, 70, 100]) 
print(my_data['my_cut'].dtypes) # category

Subhendu Mohapatra

Author & Instructor at plus2net

I write and maintain practical tutorials on Python, PHP, SQL, JavaScript, HTML, jQuery, and web development at plus2net. The tutorials focus on clear explanations, working examples, and code that readers can test and adapt while learning.

About the author   Watch on YouTube

Connect: