Как можно идиоматически запускать такую функцию, как get_dummies
, которая ожидает один столбец и возвращает несколько, в нескольких столбцах DataFrame?
Запуск get_dummies в нескольких столбцах DataFrame?
Ответ 1
Так как pandas версия 0.15.0, pd.get_dummies
может обрабатывать DataFrame напрямую (до этого он мог обрабатывать только одну серию и см. ниже обходной путь):
In [1]: df = DataFrame({'A': ['a', 'b', 'a'], 'B': ['c', 'c', 'b'],
...: 'C': [1, 2, 3]})
In [2]: df
Out[2]:
A B C
0 a c 1
1 b c 2
2 a b 3
In [3]: pd.get_dummies(df)
Out[3]:
C A_a A_b B_b B_c
0 1 1 0 0 1
1 2 0 1 0 1
2 3 1 0 1 0
Временное решение для pandas < 0.15.0
Вы можете сделать это для каждого отдельного столбца, а затем согласовать результаты:
In [111]: df
Out[111]:
A B
0 a x
1 a y
2 b z
3 b x
4 c x
5 a y
6 b y
7 c z
In [112]: pd.concat([pd.get_dummies(df[col]) for col in df], axis=1, keys=df.columns)
Out[112]:
A B
a b c x y z
0 1 0 0 1 0 0
1 1 0 0 0 1 0
2 0 1 0 0 0 1
3 0 1 0 1 0 0
4 0 0 1 1 0 0
5 1 0 0 0 1 0
6 0 1 0 0 1 0
7 0 0 1 0 0 1
Если вам не нужен стол с несколькими индексами, затем удалите keys=..
из вызова функции concat.
Ответ 2
С pandas 0,19 вы можете сделать это в одной строке:
pd.get_dummies(data=df, columns=['A', 'B'])
Columns
указывает, где делать одно горячее кодирование.
>>> df
A B C
0 a c 1
1 b c 2
2 a b 3
>>> pd.get_dummies(data=df, columns=['A', 'B'])
C A_a A_b B_b B_c
0 1 1.0 0.0 0.0 1.0
1 2 0.0 1.0 0.0 1.0
2 3 1.0 0.0 1.0 0.0
Ответ 3
У кого-то может быть что-то более умное, но здесь есть два подхода. Предполагая, что у вас есть dataframe с именем df
с столбцами "Имя" и "Год", вы хотите использовать макеты.
Во-первых, простое повторение по столбцам не так уж и плохо:
In [93]: for column in ['Name', 'Year']:
...: dummies = pd.get_dummies(df[column])
...: df[dummies.columns] = dummies
Еще одна идея - использовать пакет patsy, который предназначен для построения матриц данных из формул типа R.
In [94]: patsy.dmatrix(' ~ C(Name) + C(Year)', df, return_type="dataframe")
Ответ 4
Если я не понимаю этот вопрос, он поддерживается в get_dummies, передавая аргумент столбцов.