/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 vote278 views2 answers
Loop only to the 6th element python
Python
15 votes275 views2 answers
Delete the duplicates in python
Python
65 votes264 views6 answers
Create folder python
Python
1 vote248 views1 answer
From time import sleep, time
Python
1 vote268 views1 answer
Python source code
Python
56 votes271 views5 answers
Python format datetime
Python
0 vote249 views1 answer
How to get left click input in pygame
Python
3 votes244 views2 answers
How many columns have null values present in them? in pandas
Python
0 vote249 views1 answer
Pandas rolling mean (also known as the moving average).
Python
0 vote264 views1 answer
Custom_settings in scrpay
Python