熊猫-在列中搜索字符串并将其分类为新列

Question

让我说以下DF：

---

|Tweet|

|bla bla bad| 

|bla bla good|

---

由此，我想创建类似以下内容的

：

---

|Tweet|           |Sentiment|

|bla bla bad|     |negative|

|bla bla good|    |positive|

---

我可以使用df[df['Tweet'].str.contains("bad|sad", na=False)]在列中搜索字符串，但是如何创建新列来对它们进行分类呢？

谢谢

Answer 1

使用numpy.where：

df['sentiment'] = np.where(df['Tweet'].str.contains('bad|sad'), 'negative', 'positive')
print(df)

输出

          Tweet sentiment
0   bla bla bad  negative
1  bla bla good  positive