python - Pandas:在字符串匹配后获取列中的下一个值

Question

如何在字符串匹配之后获取列中的下一个值？

regex = 'ABC'
row_match = df["column1"].str.match(regex, case=False)

即如何在可以找到字符串 ABC 的行之后获取同一列的下一行中的值？

我尝试使用 iterrows 遍历一行，但无法让它工作。

Answer 1

您可以尝试使用 shift获取下一个值。

这是一个小例子:

>>> import pandas as pd
>>> data = ['a', 'b', 'c', 'd', 'c', 'e']
>>> df = pd.DataFrame(data)
>>> df
   0
0  a
1  b
2  c
3  d
4  c
5  e
>>> df[df[0].str.match('c').shift(1).fillna(False)]
   0
3  d
5  e

我使用 fillna替换第一行中的 NaN。