Как присвоить значения новой категории с помощью Pandas?

Question

У меня есть переменная, списки воспроизведения, которой я хочу назначить категориальную переменную, которая принимает значения «сна» или «no_sleep» для каждого списка воспроизведения. У меня есть оба списка со сном / без сна уже назначены следующим образом: Сон: «Спокойное пианино», «Сон», «Детский сон», «Белый шум»

No_sleep: «Танцевальные хиты», «Танцы» Классика »,« Массовые танцевальные хиты »,« Танцевальная вечеринка »: (« Spotify »,« 37i9dQZF1DXaXB8fQg7xif »)

Как мне добавить новую переменную в pd.dataframe и присвоить значения? Я представляю себе функцию if для сна, а все остальное - No_sleep, но не знаю, с чего начать.

Answer 1

Итак, проблема состоит в том, чтобы назначить определенные записи либо no_sleep, либо sleep с учетом имени списка воспроизведения, которое я предполагаю.

Вот один из способов сделать это:

df['sleep'] = "no_sleep"

sleep_idxs = np.array(df[df['playlist'].isin(sleep_playlists)].index)

df.iloc[sleep_idxs, 'sleep'] = "sleep"

Здесь вам понадобится список названий списков воспроизведения сна (sleep_playlists).