Как я могу настроить свой l oop, чтобы на графиках были разные заголовки - PullRequest
0 голосов
/ 26 мая 2020
er=0
for sector in range(16):
  incl = sector in wind_sectors_df
  if incl:
    ind = [i for i in range(len(wind_sectors_df)) if wind_sectors_df[i] == sector]
    if ind:  # if ind is not empty
      for i in ind:
        df = dfs[i]
        print(times[i])
        print(wind_sectors_df[i])
        latitudes = df['floatLatitudes']
        longitudes = df['floatLongitudes']
        dlat = dx * 360.0 / (2 * np.pi * earth_radius) # latitude
        dlon = dx * 360.0 / (2 * np.pi * earth_radius * np.cos(np.nanmean(latitudes) * np.pi / 180.0))
        grid_lon = np.arange(np.nanmin(longitudes), np.nanmax(longitudes), dlon) # longitude grid
        grid_lat = np.arange(np.nanmin(latitudes), np.nanmax(latitudes), dlat)   # latitude grid
        to_grid = df['CPC_tot']
        gridded_data = np.zeros([len(grid_lat), len(grid_lon)], dtype=float) + np.nan
        for j in range(len(grid_lat) - 1):
          for i in range(len(grid_lon) - 1):
            ind = np.logical_and(np.logical_and(latitudes>=grid_lat[j],  latitudes<grid_lat[j+1]),
                            np.logical_and(longitudes>=grid_lon[i], longitudes<grid_lon[i+1]))
            if np.sum(ind) >= min_n_points:
             gridded_data[j,i] = np.nanmean(to_grid[ind]) 
        xx, yy = np.meshgrid(grid_lon, grid_lat, indexing='xy')
        fig=plt.figure()
        plt.pcolor(xx, yy, gridded_data, cmap='RdBu')
        plt.colorbar(orientation='horizontal', label='CPC_tot(#/cm3)')
        plt.title('Time: {} , sector: {}'.format(times[i],sector_new_dfs[er]))
        er=er+1

Моя основная проблема в том, что все работает нормально, но когда я пытаюсь поместить заголовок в каждый отдельный сюжет, я получаю следующее сообщение:

plt.title('Time: {} , sector: {}'.format(times[i],sector_new_dfs[er]))

Ошибка:

IndexError: list index out of range 

Но если вы видите, когда я помещаю print (times [i]), он печатает все заголовки правильно ... так что я не знаю, что случилось ... когда я пытаюсь установить заголовок ... Если я удалю plt.title, весь код работает хорошо, так что это просто заголовок ... Я поместил весь код, на случай, если вам нужно что-то проверить перед plt.title ...

Я попробовал написать несколько слов в заголовке, и это сработало, просто чтобы ускорить процесс

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...