Python 如何选择与特定行匹配的行

我在下面有一个数据框

A B
a0 1
b0 1
c0 2
a1 3 
b1 4
b2 3

首先，如果df.A开始使用

“A”

，我想剪切

df

df[df.A.str.startswith（“A”）]

因此，我想削减df如下

sub1

A B
a0 1
b0 1
c0 2

A B
a0 1
b0 1

sub2

A B
a1 3 
b1 4
b2 3

A B
a1 3  
b2 3

然后，我想提取列B编号与列A开始的行A编号匹配的行。

“A”

sub1

A B
a0 1
b0 1
c0 2

A B
a0 1
b0 1

sub2

A B
a1 3 
b1 4
b2 3

A B
a1 3  
b2 3

然后追加

结果

A B
a0 1
b0 1
a1 3
b2 3

我怎样才能像这样剪切和附加

df

我尝试了

cut

方法，但效果不太好。

我认为您可以使用

mask

创建

NaN

，它由

B

值向前填充

ffill

：

请注意，对于使用

ffill