using an .loc and If statment

Question

Use the .loc method, select the column rating for the rows of df where company_location equals "U.K."

Store it in a variable called uk_ratings.

uk_ratings = choco_df.loc[:, "rating"] 
if company_location == "U.K." :
print(uk_ratings)

I am used to using SQL not Python, so a little stuck, need help in where I am going one as this doesnt runn

What does "a little stuck" mean? What *specific* problem do you have? — MattDMo, Mar 15 '23 at 15:56
Sorry should of say the above code I have writen does run and work, need to know where I am going wrong so I can fix my code — Sarah, Mar 15 '23 at 16:02

score 0 · Accepted Answer · answered Mar 15 '23 at 16:05

In pandas conditions are expressed as masks (True or False) over each entry. So, in your case, to select the rating for UK companies you should:

# 1. Consider only the rows corresponding to UK companies
uk_company_mask = choco_df['company_location'] == 'U.K.'

# 2. Take the rating column as a whole
ratings = choco_df.loc[:, 'rating']

# 3. Mask to consider only UK companies
uk_ratings = ratings[uk_company_mask]

Indeed, this can be done quite succinctly as follows:

uk_ratings = choco_df.loc[choco_df['company_location'] == 'U.K.', 'rating']

using an .loc and If statment

1 Answers1