python - 使用 pandas 选择上面有附加行的结果

Question

我正在使用以下内容在巨大的数据集中搜索具有特定字符串的行：

words= ['string1','string2']
string='|'.join(words)
mask=df.stack().str.contains(string).any(level=0)
df=df[mask]

上面的代码可以为我提供包含字符串的行。对于每个结果，我还想抓住它上面的行。如何使用 shift 来做到这一点，即，将结果和行保持在结果的正上方？

标签： pythonpandasdataframe