Additional arguments

Let's use additional arguments in the .apply() method!

Your task is to create two new columns in scores:

mean is the row-wise mean value of the math score, reading score and writing score
rank defines how high the mean score is:
- 'high' if the mean value \(> 90\)
- 'medium' if the mean value \(> 60\) but \(\leq\) 90
- 'low' if the mean value \(\leq 60\)

To accomplish this task, you'll need to define the function rank that, given a series, returns a list with two values: the mean of the series and a string defined by the aforementioned rule.

The module numpy is already imported for you as np.

Cet exercice fait partie du cours

Practicing Coding Interview Questions in Python

Afficher le cours

Instructions

Calculate the mean of the input series.
Return the mean and its rank as a list.
Insert the output of rank() into new columns of scores.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

def rank(series):
    # Calculate the mean of the input series
    mean = ____
    # Return the mean and its rank as a list
    if ____:
        return ____
    if ____:
        return ____
    return ____

# Insert the output of rank() into new columns of scores
cols = ['math score', 'reading score', 'writing score']
scores[['mean', 'rank']] = scores[cols].____
print(scores[['mean', 'rank']].head())

Modifier et exécuter le code