Skip to content Skip to sidebar Skip to footer

Extract DataFrame From A List Of Indices Of Another DataFrame

I've a DataFrame 'A' and a list of indices 'I'. I want to generate/get a DataFrame 'B' which contains only the data in those indices 'I' of the original DataFrame 'A'. How can I ac

Solution 1:

I think to need 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

Post a Comment for "Extract DataFrame From A List Of Indices Of Another DataFrame"