Pandas之query函数

我们有时需要根据条件过滤一个数据帧。过滤数据帧的一个简单方法是query函数。

import pandas as pd
import numpy as np

values_1 = np.random.randint(10, size=10)
values_2 = np.random.randint(10, size=10)
years = np.arange(2010, 2020)
groups = ['A','A','B','A','B','B','C','A','C','C']
df = pd.DataFrame({'group':groups, 'year':years, 'value_1':values_1, 'value_2':values_2})

print(df)

 

 

 

 

 

 

 

使用查询函数非常简单，只需要编写条件。

df1 = df.query('value_1 < value_2')
print(df1)