根据另一列的“组”值填充 dataframe 的列

Question

我正在尝试根据另一列的值填充列的值。 假设我有以下 dataframe：

import pandas as pd
data = {'A': [4, 4, 5, 6],
        'B': ['a', np.nan, np.nan, 'd']}
df = pd.DataFrame(data)

而且我想填充 B 列，但前提是 A 列的值等于 4。因此，与 A 列中的另一个具有相同值的所有行在 B 列中应该具有相同的值（通过填充这个）。

因此，所需的 output 应该是：

data = {'A': [4, 4, 5, 6],
        'B': ['a', a, np.nan, 'd']}
df = pd.DataFrame(data)

我知道 fillna 方法，但这给出了错误的 output 因为第三行也分配了值“A”：

df['B'] = fillna(method="ffill", inplace=True)
data = {'A': [4, 4, 5, 6],
        'B': ['a', 'a', 'a', 'd']}
df = pd.DataFrame(data)

如何获得所需的 output？

Answer 1

尝试这个：

df['B'] = df.groupby('A')['B'].ffill()

Output：

>>> df
   A    B
0  4    a
1  4    a
2  5  NaN
3  6    d