Pandas：根据从其他列中提取的 substring 截断列中的字符串（Python 3）

Question

我有一个 dataframe 有两个相关的列，“rm_word”和“article”。

数据样本：

,grouping,fts,article,rm_word
0,"1",fts,"This is the article. This is a sentence. This is a sentence. This is a sentence. This goes on for awhile and that's super ***crazy***. It goes on and on.",crazy

我想查询每篇“文章”的最后 100 个字符，以确定其行的相应“rm_word”是否出现。 如果是这样，那么我想删除出现“rm_word”的整个句子以及“文章”中出现的所有句子。

期望的结果（当“crazy”是“rm_word”时）：

,grouping,fts,article,rm_word
0,"1",fts,"This is the article. This is a sentence. This is a sentence. This is a sentence.",crazy

这个掩码能够确定一篇文章何时包含它的“rm_word”，但我在句子删除位上遇到了问题。

mask = ([ (str(a) in b[-100:].lower()) for a,b in zip(df["rm_word"], df["article"])])

print (df.loc[mask])

任何帮助将非常感激。 太感谢了。

Answer 1

这行得通吗？

df = pd.DataFrame(
    columns=['article', 'rm_word'],
    data=[["This is the article. This is a sentence. This is a sentence. This is a sentence.", 'crazy'],
          ["This is the article. This is a sentence. This is a sentence. This is a sentence. This goes on for awhile and that's super crazy. It goes on and on.", 'crazy']]
)

def clean_article(x):
    if x['rm_word'] not in x['article'][-100:].lower():
        return x
    article = x['article'].rsplit(x['rm_word'])[0]
    article = article.split('.')[:-1]
    x['article'] = '.'.join(article) + '.'
    return x


df = df.apply(lambda x: clean_article(x), axis=1)
df['article'].values

退货

array(['This is the article. This is a sentence. This is a sentence. This is a sentence.',
       'This is the article. This is a sentence. This is a sentence. This is a sentence.'],
      dtype=object)