考虑以下数组:
对于
arr = np.array(
[
[10, np.nan],
[20, np.nan],
[np.nan, 50],
[15, 20],
[np.nan, 30],
[np.nan, np.nan],
[10, np.nan],
]
)
对于
arr
中每一列中的每个单元格,我需要找到到下一个非NaN值的距离。
也就是说,预期的结果应该是这样的:expected = np.array(
[
[1, 2],
[2, 1],
[1, 1],
[3, 1],
[2, np.nan],
[1, np.nan],
[np.nan, np.nan]
]
)