时间戳

首页 > 解决方案 > 数据框根据具有相同索引的值填充空列值（使用PANDAS）

python - 数据框根据具有相同索引的值填充空列值（使用PANDAS）

问题描述

dataframe我有以下index内容tag#。我想sound根据匹配索引用正确的值填充列中的所有 NaN 值。

        pet    sound
tag#        
11      cat 
11      cat    meow
11      cat    meow
15      bird   tweet
8       pig    oink
5       dog 
8       pig 
8       pig 
8       pig 
5       dog    woof
5       dog    woof
11      cat    meow

我希望数据框看起来像这样

        pet    sound
tag#        
11      cat    meow
11      cat    meow
11      cat    meow
15      bird   tweet
8       pig    oink
5       dog    woof
8       pig    oink 
8       pig    oink 
8       pig    oink 
5       dog    woof
5       dog    woof
11      cat    meow

标签： pythonpandasdataframe

解决方案

尝试groupby apply在 0 级和ffill+bfill每组：

df = df.groupby(level=0).apply(lambda g: g.ffill().bfill())

       pet  sound
tag#             
11     cat   meow
11     cat   meow
11     cat   meow
11     cat   meow
15    bird  tweet
8      pig   oink
8      pig   oink
8      pig   oink
8      pig   oink
5      dog   woof
5      dog   woof
5      dog   woof

完整的工作示例：

import pandas as pd
from numpy import nan

# Re-create DataFrame
df = pd.DataFrame({
    'tag#': [11, 11, 11, 15, 8, 5, 8, 8, 8, 5, 5, 11],
    'pet': ['cat', 'cat', 'cat', 'bird', 'pig', 'dog', 'pig',
            'pig', 'pig', 'dog', 'dog', 'cat'],
    'sound': [nan, 'meow', 'meow', 'tweet', 'oink', nan, nan,
              nan, nan, 'woof', 'woof', 'meow']
}).set_index('tag#')

df = df.groupby(level=0).apply(lambda g: g.ffill().bfill())
print(df)

推荐阅读