Python Pandas 从另一个观察结果中输入字符串值

Question

我有以下代码，因此我需要从 data_current 到达 data_needed。我想我必须使用groupby，smth。像:

data_needed=data_current['Code'].fillna(data_current.groupby('animal')['Code'])

但它并没有真正起作用......

import pandas as pd
import numpy as np

data_current=pd.DataFrame({'animal':['rabbit','badger', 'hamster','badger'],'Code':['I-1','I-2','I-3',np.nan]})
data_current

data_needed=pd.DataFrame({'animal':['rabbit','badger', 'hamster','badger'],'Code':['I-1','I-2','I-3','I-2']})
data_needed

顺便问一下，如何将 i-Python 笔记本单元格/输出粘贴到 Stackoverflow 问题中，以便每个人都可以直接在此页面上看到输出？

Answer 1

您可以对“animal”列进行分组，然后在“animal”列上调用transform，只需调用fillna并传递参数method='ffill' 将填充任何 NaN 值:

In [38]:

data_current['code'] = data_current.groupby('animal')['code'].transform(lambda x: x.fillna(method='ffill'))
data_current
Out[38]:
    animal code
0   rabbit  I-1
1   badger  I-2
2  hamster  I-3
3   badger  I-2