Интеграция в корреляционную матрицу некоторых данных

У меня есть данный код, который строит корреляционную матрицу между индексом MJO, мне нужно добавить в код ещё данные другого индекса, что бы в матрице ещё была корреляция с новым индексом Вот код которым я строю карты

from math import ceil, floor
import xarray as xr
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors
from matplotlib import ticker, cm
from matplotlib.ticker import FuncFormatter
import cartopy.crs as ccrs
from cartopy.mpl.ticker import (LongitudeFormatter, LatitudeFormatter,
                                LatitudeLocator)

from scipy import stats as st

import matplotlib.path as mpath
import cartopy.feature as cfeature
import matplotlib.ticker as mticker
import pandas as pd
import seaborn as sns

y1 = 1979

y0_t = 1978
yn_t = 2021

months = (['Jan','Feb','Mar','Apr','May','Jun','Jul','Aug','Sep','Oct','Nov','Dec'])
mon_ = (['01','02','03','04','05','06','07','08','09','10','11','12'])
mjo_ = (['01','02','03','04','05','06','07','08','09','10'])
mjo_lon = (['80E','100E','120E','140E','160E','120W','40W','10W','20E','70E'])

mon = 0

years_t = range(y0_t,yn_t+1)
nyear_t = len(years_t)

i_mjo = 0

mjo = np.zeros((10, nyear_t, 12), dtype=float)

for i_mjo in range(10):
    fi = 'D:/diplom/mjoCode/icpc_mjo' + mjo_[i_mjo] + '_mean12.dat.txt' # mjo_[i_mjo]
    data = np.loadtxt(fi) # (44, 13)
    mjo[i_mjo] = data[:,1:13]
i=0

for mon in range(12):
    df = pd.DataFrame(mjo[:,:,mon].T, columns=mjo_lon) #['assists','rebounds','points']
    fig = sns.heatmap(df.corr().round(2), annot = True)
    plt.figure()#figsize=(16, 6)
    heatmap = sns.heatmap(df.corr().round(2), vmin=-1, vmax=1, annot=True, cmap ='coolwarm')#cmap='BrBG'
    heatmap.set_title('Correlation MJO  '+ months[mon]+ '  ' + str(y0_t) + '-' + str(yn_t), fontdict={'fontsize':12}, pad=12);
    plt.savefig('D:/diplom/mjoMap/heatmap_' + mon_[mon]+'_'+ months[mon]  +'.jpg', dpi=300, bbox_inches='tight')

Данные которые я использую можно найти тут https://climexp.knmi.nl/data/icpc_mjo01_mean12.dat , и вот данные которые я хочу добавить https://climexp.knmi.nl/data/isnao_ucar.dat . Код не мой, поэтому не могу решить данную проблему


Ответы (0 шт):