pd column to one hot vector

Attractive Alpaca answered on February 2, 2022 Popularity 1/10 Helpfulness 1/10

answer pd column to one hot vector

pd column to one hot vector

Comment

import pandas as pd
        
df = pd.DataFrame({
          'A':['a','b','a'],
          'B':['b','a','c']
        })
df
Out[]: 
   A  B
0  a  b
1  b  a
2  a  c

# Get one hot encoding of columns B
one_hot = pd.get_dummies(df['B'])
# Drop column B as it is now encoded
df = df.drop('B',axis = 1)
# Join the encoded df
df = df.join(one_hot)
df  
Out[]: 
       A  a  b  c
    0  a  0  1  0
    1  b  1  0  0
    2  a  0  0  1

Popularity 1/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: python vector

Link to this answer
Share Copy Link