如何使用 Pandas 將值分配給新類別？

Question

我有一個變量，播放列表，我想為每個播放列表分配一個分類變量，該變量采用“sleep”或“no_sleep”的值。 我有兩個睡眠/沒有睡眠的列表已經分配如下：睡眠：“和平鋼琴”，“睡眠”，“嬰兒睡眠輔助”，“白噪聲”

No_sleep：“Dance Hits”、“Dance Classics”、“Massive Dance Hits”、“Dance Party”：（“Spotify”、“37i9dQZF1DXaXB8fQg7xif”）

如何將新變量添加到 pd.dataframe 並分配值？ 我想如果 function 用於睡眠，其他一切都是 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 ）。