Извлечь DataFrame из списка индексов другого DataFrame

Question

Извлечь DataFrame из списка индексов другого DataFrame

1

У меня есть DataFrame "A" и список индексов "I". Я хочу создать/получить DataFrame "B", который содержит только данные в этих индексах "I" исходного DataFrame "A". Как я могу это достичь?

Предполагая, что I = [1, 3], я попробовал этот A.filter(items=I, axis=0), это правильный путь или есть еще лучший способ сделать это.

Temp O'rary 25 сен. 2018, в 10:12

Источник

Теги:

python

pandas

dataframe

sklearn-pandas

1 ответ

Ещё вопросы

jezrael · Accepted Answer · 2018-09-25T05-03-00.000Z

Мне кажется, нужен 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