在python中找到nan的连续计数？

import numpy as np

# 创建一个包含nan的示例数组
arr = np.array([1, 2, np.nan, np.nan, 5, np.nan, 7, 8, np.nan])

# 使用np.isnan函数判断数组中的元素是否为nan
nan_indices = np.isnan(arr)

# 使用np.diff函数计算相邻元素的差异
diff_indices = np.diff(nan_indices)

# 使用np.where函数找到连续的nan序列的起始和结束索引
start_indices = np.where(diff_indices == 1)[0]
end_indices = np.where(diff_indices == -1)[0]

# 打印连续nan序列的起始和结束索引
for start, end in zip(start_indices, end_indices):
    count = end - start + 1
    print(f"连续的nan序列起始索引：{start}，结束索引：{end}，计数：{count}")