根据数据帧的索引保留另一个数据帧中的索引

Question

>假设我有两个数据帧A，并且B从0到10索引。我从 A 中删除了几个重复的行，以便删除索引7和9。所以现在A.index将是[0,1,2,3,4,5,6,8,10].

现在我想在B中准确地保留具有这些相同索引的行。截至目前，其索引集为从0到10。换句话说，最初给定完全相同的索引，并且从A中删除了一些索引，我如何保留B行的子集，这些子集与保留的A行在其索引方面完全对应？

Answer 1

我相信你可以按loc选择：

A = pd.DataFrame({'col':[5,8,4,0,6,2,1,8,3,4,9]})
B = pd.DataFrame({'col':np.arange(10, 21)})
#print (A)
#print (B)
A1 = A.drop_duplicates('col')
print (A1)
col
0     5
1     8
2     4
3     0
4     6
5     2
6     1
8     3
10    9
B1  = B.loc[A1.index]
print (B1)
col
0    10
1    11
2    12
3    13
4    14
5    15
6    16
8    18
10   20