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

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