Pandas —在列中搜索字符串，并将其分类在新列中[duplicate]

Question

此问题已经在这里有了答案：

• Pandas conditional creation of a series/dataframe column7个答案

我想说以下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