[英]Extract DataFrame from a list of indices of another DataFrame
我有一个DataFrame“A”和一个索引列表“I”。 我想生成/获取一个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
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.