Python/Pandas 在 Dataframe 中搜索数据

Question

我想用一个例子来解释我的问题。我有一个数据集，其中包括鳄梨平均价格和这些价格的许多特征（我想鳄梨价格数据集非常受欢迎，idk）。并且有一个称为“区域”的特征可以显示鳄梨的生长地。我编写这行代码是为了获得在“西部”生长的鳄梨功能。我的数据名称是数据 btw

west = data[data['region'] =='West'] 

而且我想，如果我想获得 2016 年生长并且也在西部生长的鳄梨怎么办。如何同时获取这些数据？

Answer 1

您可以尝试 pandas 的 query 接口。

特别是，如果您的“成长”数据出现在 year 列中，您可以做类似的事情，

data.query('region == "West" and year == 2016')

参考文献：

• https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.query.html
• https://www.geeksforgeeks.org/python-filtering-data-with-pandas-query-method/

Answer 2

我认为 pandas 具有布尔条件的 DataFrame 过滤器可以解决您的问题。

假设鳄梨种植年份的列名称是 grew_in。然后试试这个：

west_2016 = data[(data['region'] =='West') & (data['grew_in'] == 2016)] 

Answer 3

west = data[(data['region']=='west')&(data['year']==2016)]