根據 Pandas DataFrame 檢查列表中的值

Question

我有一個 Pandas DataFrame，其中包含單詞及其相關情緒的列表（每個單詞都可以附加多種情緒）。 是這樣的：

我還使用 Spacy 將文本的標記提取到列表中。 像 ['study', 'Maths', 'easy', 'great', 'study',...]

為了將 tokenList 中的標記與情感 dataframe (df_lexicon) 中的相關情感相匹配，我嘗試了以下操作：`

emotions = []

// adding the token to emotions list if it exists in the emotion dataframe

for i in tokensList:
  if i in df_lexicon['word'].values:
    emotions.append(i)

// printing the row including the word and emotion

for i in emotions:
  print(df_lexicon[df_lexicon['word']==i])

But that gives me:

       word   emotion
10215  ban  negative
       word   emotion
10220  mad    negative
       mad    fear
.
//(and many more)

我不知道如何將結果添加到新的 DataFrame 而不是打印它們。 感謝你的幫助。

Answer 1

您可以使用.isin()對照列表中的值檢查 dataframe：

s = df_lexicon['word'].isin(tokenList)

new_df = df_lexicon[s]

如果它解決了您的問題，請告訴我。