从另一个 DataFrame 的索引列表中提取 DataFrame
Extract DataFrame from a list of indices of another DataFrame
我有一个 DataFrame "A" 和一个索引列表 "I"。我想要 generate/get 一个 DataFrame "B"，它只包含原始 DataFrame "A" 的那些索引 "I" 中的数据。我怎样才能做到这一点？
假设 I = [1, 3] ，我试过 A.filter(items=I, axis=0) 这是正确的方法，还是有更好的方法。
我认为需要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
我有一个 DataFrame "A" 和一个索引列表 "I"。我想要 generate/get 一个 DataFrame "B"，它只包含原始 DataFrame "A" 的那些索引 "I" 中的数据。我怎样才能做到这一点？
假设 I = [1, 3] ，我试过 A.filter(items=I, axis=0) 这是正确的方法，还是有更好的方法。
我认为需要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