Skip to content

Bees Knees Solution


knees.loc[pd.isna(bees).to_numpy()] *= 2

print(knees)
# 7     5
# 0     2
# 2     9
# 6     2
# 3     3
# 5    10
# 1     5
# 4     4
# dtype: int64

Explanation

  1. Identify which elements of bees are NaN with pd.isna().

    pd.isna(bees)
    # 0    False
    # 1    False
    # 2    False
    # 3     True
    # 4    False
    # 5    False
    # 6    False
    # 7     True
    # dtype: bool
    

    This results in a boolean Series.

  2. Suppose we use this boolean Series to index knees..

    knees.loc[pd.isna(bees)]
    # 7    5
    # 3    3
    # dtype: int64
    

    The elements it selects are not the ones we want. knees.loc[pd.isna(bees)] selects the elements of knees whose index matches those of pd.isna(bees) with True values. In other words, pd.isna(bees) has True values at indexes 3 and 7, so knees.loc[pd.isna(bees)] selects rows of knees with index 3 and 7.

    However, the problem asked use to double the ith value of knees if the ith value of bees was NaN. The fourth and eighth elements of bees are NaN, so we want to select the fourth and eight elements of knees. (Note that knees's index is not in sequential order.)

    To prevent Pandas from using the index of pd.isna(bees) to select elements of knees, we can remove the index by converting the boolean Series to a NumPy array. Then Pandas will select elements by the position of True values.

    knees.loc[pd.isna(bees).to_numpy()]
    # 6    1
    # 4    2
    # dtype: int64
    
  3. Double the value of those elements.

    Easy enough, thanks to the += operator. 😃

    knees.loc[pd.isna(bees).to_numpy()] *= 2
    
    print(knees)
    # 7     5
    # 0     2
    # 2     9
    # 6     2
    # 3     3
    # 5    10
    # 1     5
    # 4     4
    # dtype: int64
    

See the problem