Методы получения фиктивных переменных в кадре данных Pandas

Чтобы получить фиктивные данные в DataFrame Pandas, вы можете использовать функцию pd.get_dummies(). Эта функция преобразует категориальные переменные в фиктивные/индикаторные переменные. Вот несколько методов, которые вы можете использовать:

  1. Использование функции pd.get_dummies():

    import pandas as pd
    # Create a DataFrame
    df = pd.DataFrame({'Category': ['A', 'B', 'C', 'A', 'B', 'C']})
    # Get dummies
    dummies = pd.get_dummies(df['Category'])
    df = pd.concat([df, dummies], axis=1)
  2. Использование конструктора pd.DataFrame():

    import pandas as pd
    # Create a DataFrame
    df = pd.DataFrame({'Category': ['A', 'B', 'C', 'A', 'B', 'C']})
    # Get dummies using pd.DataFrame()
    dummies = pd.DataFrame(pd.Categorical(df['Category']).codes, columns=['Category'])
    df = pd.concat([df, pd.get_dummies(dummies)], axis=1)
  3. Использование pd.concat()с pd.get_dummies():

    import pandas as pd
    # Create a DataFrame
    df = pd.DataFrame({'Category': ['A', 'B', 'C', 'A', 'B', 'C']})
    # Get dummies using pd.concat()
    dummies = pd.get_dummies(df['Category'])
    df = pd.concat([df, dummies], axis=1)