/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

0 vote177 views1 answer
Legend text color matplotlib
Python
0 vote190 views1 answer
Pd sample every class
Python
8 votes175 views1 answer
Python get keypressed value
Python
19 votes200 views6 answers
How to import turtle in python
Python
0 vote185 views1 answer
Pandas iat 0
Python
0 vote195 views1 answer
Numpy randomly swap lines
Python
0 vote177 views1 answer
"'S3' object has no attribute 'Bucket'", python boto3 aws
Python
0 vote224 views2 answers
How to print the freq of each char by using dict in python
Python
3 votes183 views2 answers
Getting text input pygame python
Python
1 vote171 views1 answer
Code hand tracking
Python