Get data from one column based on another using Python

Question

I'm trying to get data from column1 of a .xlsx file, when the data I provide matches the data in column2 and column4. I tried using data[data['column2'] == "aa1"] but not satisfied with the result. For column4 data, I used int(raw_input("get data: ")) but that didn't work either.

Sample excel file:

column1,column2,column3,column4
aa,aa1,lll,21
bb,aa2,ll,22

Required output:

aa #when an input of aa1 and 21 is given.

PS- The data is in the form of a pandas dataframe.

@dbliss I did excel = pd.ExcelFile(file_name) df = excel.parse("Sheet1") — pam
– pam, Commented Apr 15, 2015 at 7:35
@AndyHayden It prints the complete row, when I just want data from column1. I tried suffixing with [0], that gave me a keyerror. — pam
– pam, Commented Apr 15, 2015 at 7:36
ok, so you're asking how to get data in a pandas DataFrame. please edit your question to include this. — abcd
– abcd, Commented Apr 15, 2015 at 7:37

EdChum · Accepted Answer · 2015-04-15 07:54:06Z

1

You want this:

data[(data['column2'] == "aa1") & (data['column4'] == 21)]

So to use multiple conditions you need to use & instead of and because we are comparing arrays also you need to use parentheses due to operator precedence

answered Apr 15, 2015 at 7:54

EdChum

397k204 gold badges836 silver badges583 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Get data from one column based on another using Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related