python - Pandas :drop_duplicates().fillna(0) 不填充零

Question

我正在尝试从系列中删除重复项，然后用 0 填充 NaN。应该非常简单，但是当我将这两个函数链接在一起时，我仍然会在我期望的位置得到 NaN 0.0。

df = pd.DataFrame({'a':[1,1,2,3,3,4], 'b':[10,20,30,40,50,60]})
df['a'] = df['a'].drop_duplicates().fillna(0)

产量:

     a   b
0  1.0  10
1  NaN  20
2  2.0  30
3  3.0  40
4  NaN  50
5  4.0  60

鉴于:

df['a'] = df['a'].drop_duplicates()
df['a'] = df['a'].fillna(0)

工作正常，但似乎不必要地冗长。

我正在使用 pandas 0.20.3。

Answer 1

因为 drop_duplicates 删除了元素，所以您的结果比您分配回的数据框更小。这会导致索引不匹配。

或者，使用 mask/where 和 duplicated -

df['a'] = df['a'].mask(df['a'].duplicated(), 0)

或者，

df['a'] = df['a'].where(~df['a'].duplicated(), 0)

df

   a   b
0  1  10
1  0  20
2  2  30
3  3  40
4  0  50
5  4  60