python 使用.loc和If语句

33qvvth1  于 2023-03-16  发布在  Python
关注(0)|答案(1)|浏览(146)

enter image description here
使用.loc方法,为company_location等于“U.K.”的df行选择列评级。
将其存储在名为uk_ratings的变量中。

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

我习惯使用SQL而不是Python,所以有点卡住了,需要帮助,在我要去的地方,因为这不运行

mwkjh3gx

mwkjh3gx1#

在Pandas中,每个条目的条件都表示为 masksTrueFalse)。因此,在您的情况下,要选择英国公司的评级,您应该:

# 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]

实际上,这可以非常简洁地做到如下:

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

相关问题