if not np.isnan(heights[j]) and not np.isnan(heights[j + time_interval]):
    axe_x.extend(time_list[np.arange(j + n, j + (time_interval - n))])
TypeError: только целочисленные массивы с одним элементом могут быть преобразованы в индекс 3