создать новый df со значением из списка

#python #pandas #dataframe

#python #pandas #фрейм данных

Вопрос:

У меня есть df и список значений, и я хочу создать новый df только со значением из списка. Что мне делать?

 df = pd.DataFrame({'A' : [5,6,3,4,6], 'B' : [1,2,3, 5,1]}) 
list_of_values = [3,6] 
result= df.query("A in @list_of_values")
print(result)
  

Это то, что у меня есть:

    A  B
1  6  2
2  3  3
4  6  1
  

Это то, что я хочу:

    A  B
0  6  2
1  3  3
2  6  1
  

Комментарии:

1. Вы можете reset_index , но вы уверены, что хотите?

Ответ №1:

Просто сделайте drop=True под reset_index() :

 result= df.query("A in @list_of_values").reset_index(drop=True)
  

 df = pd.DataFrame({'A' : [5,6,3,4,6], 'B' : [1,2,3, 5,1]}) 
list_of_values = [3,6] 
result= df.query("A in @list_of_values").reset_index(drop=True)
print(result)

   A  B
0  6  2
1  3  3
2  6  1