python筛选出csv满足某条件的行,如何通过在python中的某些列上应用条件来过滤csv数据...

I am new python data analysis and having some problems to get the required data in specific format.

My data is in following format. ( please check the attached link for data in csv format as the data is quite large)

I used following commands to print the csv data in the above format

address = 'C:\Barchatdata.csv'

data_c = pd.read_csv(address)

Now i want to apply if condition on Energy_Supply_per_capita >280 and then print index column, contry_area, Energy_Supply_per_capita and Avg_GDP columns.

i tried following command

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

but got only index and Energy_Supply_per_capita columns.

How i can get the required results?

Thank you in advance.

解决方案

You can use query

cols = ['Country_Area', 'Energy_Supply_per_capita', 'Avg_GDP']

data_c.query('Energy_Supply_per_capita > 280')[cols]

Or equivalently with a boolean series and loc

cols = ['Country_Area', 'Energy_Supply_per_capita', 'Avg_GDP']

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