/Python/Impute data by using groupby and transform

Impute data by using groupby and transform

import pandas as pd
from datetime import datetime

def generate_data():
    ...

t = datetime.now()
df = generate_data()
df['value'] = df.groupby(['category', 'name'])['value']\
    .transform(lambda x: x.fillna(x.mean()))
print(datetime.now()-t)

# 0:00:00.016012

t = datetime.now()
df = generate_data()
df["value"] = df.groupby(['category', 'name'])\
    .transform(lambda x: x.fillna(x.mean()))['value']
print(datetime.now()-t)

# 0:00:00.030022

Source: stackoverflow.com

Impute data by using groupby and transform

df['value'] = df.groupby(['category', 'name'])['value']\
    .transform(lambda x: x.fillna(x.mean()))

Source: stackoverflow.com

Similar codes

3 votes214 views2 answers
Tensorflow
Python
1 vote214 views1 answer
Location of last row dataframe
Python
0 vote206 views1 answer
Reveal a defined function in python
Python
0 vote191 views1 answer
Python long numbers as string
Python
0 vote208 views2 answers
How to take unknown number of inputs in python
Python
12 votes207 views10 answers
Fortnite
Python
0 vote199 views1 answer
Pandas remove item from dictionary
Python
0 vote210 views1 answer
How to create pyw file
Python
-1 vote203 views1 answer
Email validation using django
Python
0 vote216 views1 answer
Rstudi matplotlib crash qt
Python