Python中類似Grepl(R)函數的用法

Question

我正在嘗試用 Python 編寫以下代碼，我已經用 R 編寫了它。這很簡單，但是我對 Python 非常陌生。 如果df$text 中存在任何單詞（來自單詞列表），我會將text列指定為 1 到pos_d列中。 任何幫助，將不勝感激。

wordlist<-list("word1|word2")
df$pos_d <- grepl(wordlist, df$text)
df$pos_d [df$pos_d == "true"] <- 1

Answer 1

干得好

如果 wordlist 是一個列表

df['pos_d'][(df['text'].isin(wordlist)] = 1 

如果 wordlist 是一個字符串

df['pos_d'][(df['text'] in (wordlist)] = 1 

享受並繼續發布！

Answer 2

你也可以使用re和str.contains來做到這str.contains ：

import re
boolean_mask_rows = df['text'].astype(str).str.contains(wordlist, regex = True, case = False)
df.loc[boolean_mask_rows, 'pos_d'] = 1

以上，

在第一行中，我們根據 wordlist 中的匹配字符串與 df 中的“text”列來獲取布爾向量。

在第二行中，我們使用帶有loc布爾向量索引來寫 1。