为什么 fillna 在 pandas 2.1.4 版本中没有按预期工作？

Question

这是我的数据框：

import pandas as pd 
df = pd.DataFrame(
    {
        'a': ['long', 'long', 'short', 'long', 'short', 'short', 'short'],
        'b': [1, -1, 1, 1, -1, -1, 1],
    }
)

预期输出正在创建列

a_1

：

        a    b       a_1
0     long   1       long
1     long  -1       long
2    short   1      short
3     long   1       long
4    short  -1       long
5    short  -1       long
6    short   1      short

逻辑：

a_1

应该像这样创建：

df.loc[df.b.eq(-1), 'a_1'] = 'long'
df['a_1'] = df.a_1.fillna(df.a)

这个问题真的很奇怪。当我尝试

fillna

时，它不起作用。我在 pandas 版本 1.2.4 上尝试过，它可以工作，但在版本 2.1.4 上它不起作用。 该版本是当前 Colab 的默认版本，我在 Colab 上运行了此代码。

Answer 1

在 pandas 2.2.2 版本中工作正常，这可能是该版本中的一个错误。

但是你可以像这样实现相同的输出：

df['a_1'] = df.apply(lambda row: 'long' if row['b'] == -1 else row['a'], axis=1)