从另一个DataFrame的索引列表中提取DataFrame

Question

我有一个名为"A"的DataFrame和一个索引列表"I"。我想生成/获取一个名为"B"的DataFrame，其中只包含原始DataFrame"A"中这些索引"I"的数据。我该如何实现这一点？

假设"I = [1, 3]"，我尝试了这个方法"A.filter(items=I, axis=0)"，这是正确的方法吗？还是有更好的方法？

- Temp O'rary

Answer 1

我认为需要使用 DataFrame.loc 方法：

A = pd.DataFrame({
    'A': ['a','a','a','a','b','b','b','c','d'],
    'B': list(range(9))
})
print (A)
   A  B
0  a  0
1  a  1
2  a  2
3  a  3
4  b  4
5  b  5
6  b  6
7  c  7

I = [1,3]
B = A.loc[I]
print (B)
   A  B
1  a  1
3  a  3

- jezrael