我有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
只需执行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
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句