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

【问题标题】：Extract DataFrame from a list of indices of another DataFrame从另一个 DataFrame 的索引列表中提取 DataFrame
【发布时间】：2019-02-28 18:25:41
【问题描述】：

我有一个数据框“A”和一个索引列表“I”。我想生成/获取一个数据帧“B”，它只包含原始数据帧“A”的那些索引“I”中的数据。我怎样才能做到这一点？

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

【问题讨论】：

标签： python pandas dataframe sklearn-pandas

【解决方案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

【讨论】：