форма столбца меток является непоследовательной после разделения данных на элементы и метки

Question

У меня есть набор данных в виде панды, который нужно разделить на набор функций и метки.на данный момент я делю столбцы, как показано ниже,

features = df2.drop('case_of_injury_group', 1)
labels = df2['case_of_injury_group']

, но форма меток не такая, как я ожидал,

features.shape

дают (39778, 12) и

labels.shape

дает (39778,), но я хочу это как (39778, 1).Пожалуйста, дайте мне знать, что я делаю не так здесь

Answer 1

Если хотите один столбец DataFrame, выберите по одному элементу вложенного списка:

labels = df2[['case_of_injury_group']]