Skip to content

Instantly share code, notes, and snippets.

@grahamharrison68
Created November 24, 2021 07:42
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Embed
What would you like to do?
df_drug = pd.read_csv("drug200.csv")
label_encoder = LabelEncoder()
categorical_features = [feature for feature in df_drug.columns if df_drug[feature].dtypes == 'O']
for feature in categorical_features:
print(feature, list(df_drug[feature].unique()), list(label_encoder.fit_transform(df_drug[feature].unique())), "\n")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment