我应该如何构造一个采用 pandas dataframe 并写入其列的 function 并写入其列？

Question

I want to write a function that iterates through a dataframe, and takes each row's value as an argument. For example:

My pandas dataframe is as follows:

category  sales  met_sales
fruit     100    False
books     200    False
fruit     300    False

I have a dictionary: required_sales = {'fruit':150, 'books':200}

The output I want is this:

category  sales  met_sales
fruit     100    False
books     200    True
fruit     300    True

Is it correct to structure my function like that?

def met_sales(df, dict):
    for row in df:
        if row.sales > required_sales[row.category]:
             #update met_sales column
             row.met_sales = True

Then, I can simply call met_sales(df,required_sales) to update my DataFrame.

Is this a good way of using self created functions to modify my DataFrame?

Answer 1

Use Series.map for dictionary and compare with column sales :

df['met_sales'] = df['sales'] >= df['category'].map(required_sales)
print (df)
  category  sales  met_sales
0    fruit    100      False
1    books    200       True
2    fruit    300       True

Detail :

print (df['category'].map(required_sales))
0    150
1    200
2    150
Name: category, dtype: int64

Function:

Dont use dict as variable name as it is a reserved word for builtin python dict .

def met_sales(df, d):
    df['met_sales'] = df['sales'] >= df['category'].map(d)
    return df

df1 = met_sales(df,required_sales)
print (df1)
  category  sales  met_sales
0    fruit    100      False
1    books    200       True
2    fruit    300       True

Notice :

It is necessary that all values of category are present in your dict, else missing values are returned for not existing keys:

required_sales = {'fruit':150}

print (df['category'].map(required_sales))
0    150.0
1      NaN
2    150.0
Name: category, dtype: float64