python - numpy 数组中的反向计数

Question

假设我有一系列整数值排列在像这样的 numpy 数组中。

nan = np.nan
arr = np.array([3, nan, nan, nan, 5, nan, nan, nan, nan, nan])

nan 值应该用从第一个非空值到零的反向计数填充。

[3, 2, 1, 0, 5, 4, 3, 2, 1, 0]

Answer 1

IMO，最简单的 pandas 方法是使用 groupby 和 cumcount 以及 ascending=False:

s = pd.Series(np.cumsum(~np.isnan(arr)))
s.groupby(s).cumcount(ascending=False)

0    3
1    2
2    1
3    0
4    5
5    4
6    3
7    2
8    1
9    0
dtype: int64