Skip to content

Instantly share code, notes, and snippets.

@petrosDemetrakopoulos

petrosDemetrakopoulos/pandas_categorical_1.py

Created September 17, 2022 23:31
Show Gist options
  • Save petrosDemetrakopoulos/8567099778ac47f45e2d45174cfe10df to your computer and use it in GitHub Desktop.
Save petrosDemetrakopoulos/8567099778ac47f45e2d45174cfe10df to your computer and use it in GitHub Desktop.
Download ZIP
Pandas categorical column as object
Raw
pandas_categorical_1.py
import pandas as pd
import random
colors = ['WHITE', 'BLACK','RED','YELLOW','BLUE','GREEN']
colors_column = [random.choice(colors) for x in range(1000)]
df = pd.DataFrame({'color':colors_column})
print(df.info(memory_usage="deep"))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment