Аргумент должен быть строкой или числом [Кодировка метки] - PullRequest
0 голосов
/ 30 мая 2020

Я пытаюсь добавить свой фрейм данных в новый фрейм данных, но получаю ошибку «Аргумент должен быть строкой или числом».

# The encoders
le = LabelEncoder()
ohc = OneHotEncoder()
for col in num_ohc_cols.index:

   # Integer encode the string categories
   dat = le.fit_transform(df_ohc[col]).astype(np.int)
   # Remove the original column from the dataframe
    df_ohc = df_ohc.drop(col,axis=1)
   # One hot encode the data--this returns a sparse array
   new_dat = ohc.fit_transform(dat.reshape(-1,1))
   # Create unique column names

   n_cols = new_dat.shape[1]

   col_names = ['_'.join([col,str(x)]) for x in range(n_cols)]
   print(col_names)
   # Create the new dataframe

Я получаю сообщение об ошибке при создании нового фрейма данных:

new_df=pd.DataFrame(
 new_dat.toarray(),index=df_ohc.index,columns=col_names)

Ответы [ 2 ]

0 голосов
/ 30 мая 2020

Я решил, добавив изменение моего метода добавления как:

df_ohc = pd.concat([df_ohc, new_df], axis=1)

0 голосов
/ 30 мая 2020

Эта ошибка вызвана тем, что в ваших данных действительно есть как число, так и строка. Лучший способ исправить это - преобразовать все данные в строку следующим образом:

new_df = new_df.apply(lambda x: le.fit_transform(x.astype(str)), axis=0, result_type='expand')
...