Use 'loc' to select multiple columns

Question

I have a data set of news with 3 columns: lead_paragraph, _id, web_url.

The following code returns a row matching the id along with News lead_paragraph.

ds = pd.read_csv("nytimes.csv")
def item(id):
    return ds.loc[ds['_id'] == id]['lead_paragraph'].tolist()[0]

How to get web_url also with lead_paragraph in the list?

Got an error after trying one solution

return ds.loc[ds['_id'] == id], ['web_url', 'lead_paragraph']].tolist()
                                                             ^
SyntaxError: invalid syntax

return [ds['_id'] == id], ['lead_paragraph','web_url']].tolist()[0] Its giving syntax error @coldspeed — carl
– carl, Commented Mar 28, 2019 at 20:50
` return ds.loc[ds['_id'] == id], ['lead_paragraph','web_url' ] ].tolist()[0] ` Syntax error again — carl
– carl, Commented Mar 28, 2019 at 21:00

Scott Boston · Accepted Answer · 2019-03-28 21:26:50Z

2

Use df.loc[row indexer, columns indexer],

df.loc[df['_id'] == id, ['web_url', 'lead_paragraph']].values[0].tolist()

Where row indexer is a boolean series and column indexer is a list of column labels.

edited Mar 28, 2019 at 21:26

answered Mar 28, 2019 at 20:51

Scott Boston

154k15 gold badges160 silver badges207 bronze badges

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

3 Comments

carl Over a year ago

return ds.loc[ds['_id'] == id], ['lead_paragraph','web_url']].tolist()[0] its giving me syntax error

Scott Boston Over a year ago

You are correct, there was an extra ] before the first comma.

carl Over a year ago

Can i refer my whole code , so that you can see what i am trying to do

Collectives™ on Stack Overflow

Use 'loc' to select multiple columns

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related