扫描表以查找列中的某个值，然后返回另一列 Python 的先前值 [关闭]

Question

使用 python，从这样的表中：

A  B  C
0  1  5
1  1  8
2  1  5
3  1  4
4  1  5
5  5  10
6  5  9
7  1  5
8  1  4
9  1  5
10 5  4

如何创建一个函数来扫描以查找所有“B”值 = 5，然后返回前 4 个“C”值，其中“B” = 1？如果只有 3 个 ‘B’ = 1，则返回 3。

在第一个 'B' = 5 之后，我需要输出 8、5、4、5。第二个 'B' = 5 应该什么都不返回，因为它前面没有任何 'B' = 1 .第三个 ‘B’ = 5 需要返回 5, 4, 5。这应该被写入一个全新的数据集中，每个新的 ‘B’ = 5 值之间都有一个空格，如下所示：

8
5
4
5

5
4
5

Answer 1

选项 1
loc + groupby + tail -

i = df.B.eq(5).cumsum()
df.loc[df.B == 1, 'C'].groupby(i, group_keys=False).tail(4)

1    8
2    5
3    4
4    5
7    5
8    4
9    5
Name: C, dtype: int64

---

选项 2
groupby + apply。

v = df.B.groupby(i, group_keys=False).apply(lambda x: x[x == 1].iloc[-4:]).index    
df.loc[v, 'C']

1    8
2    5
3    4
4    5
7    5
8    4
9    5
Name: C, dtype: int64