Search code examples
pythonpandasdummy-variable

Create dummy variable column from value column

I know that Pandas has a get_dummy function which you can use to convert categorical variables to dummy variables in a DataFrame. What I'm trying to do is slightly different.

I have a column containing percentage values from 0.0 to 100.0. I need to convert this to a column that has 1's for any value >= 10.0 and 0's for any value < 10.0. Is there a good way to do this repurposing get_dummy here or will I have to construct a loop to do it?

Solution

  • You can can convert bools to ints directly:

    (df.column_of_interest >= 10).astype(int)