Dataframe 如果在 dataframe 列上满足条件，那么在 dataframe 中获取这一行和接下来的 3 行？

Question

我想编写一个代码，以便如果在 dataframe 列上满足条件，那么我可以在 dataframe 中获得这一行和接下来的 3 行？

每次我的资金费率大于 0.0363 时，我都想要该行和匹配行的下 3 行。

Answer 1

您可以使用内置模块functools中的reduce来动态执行此操作：

import functools

N = 3
cond = df['Funding Rate'].str.strip('%').astype(float).gt(0.0363)
df[functools.reduce(lambda x, y: x | y, [cond.shift(i) for i in range(N + 1)])]