如何通过在python中的某些列上应用条件来过滤csv数据

Question

我是新的python数据分析，并且在获取特定格式的所需数据时遇到一些问题。

我的数据采用以下格式。 （由于数据很大，请检查附件链接是否有csv格式的数据）

我使用以下命令以上述格式打印csv数据

address = 'C:\\Barchatdata.csv' data_c = pd.read_csv(address)

现在，我想应用Energy_Supply_per_capita> 280上的条件，然后打印索引列，contry_area，Energy_Supply_per_capita和Avg_GDP列。

我尝试了以下命令

data_c.loc[data_c['Energy_Supply_per_capita'] > 280, 'Energy_Supply_per_capita']

但只有index和Energy_Supply_per_capita列。

我如何获得所需的结果？

先感谢您。

链接到csv文件

Answer 1

您可以使用query

cols = ['Country_Area', 'Energy_Supply_per_capita', 'Avg_GDP']
data_c.query('Energy_Supply_per_capita > 280')[cols]

或等效地具有布尔序列和loc

cols = ['Country_Area', 'Energy_Supply_per_capita', 'Avg_GDP']
data_c.loc[data_c.Energy_Supply_per_capita > 280, cols]