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