#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