Корреляция между массивами разных форм в питоне - PullRequest
Новая
       9

Корреляция между массивами разных форм в питоне

0 голосов
/ 07 октября 2019

У меня есть массив 4 * 4 с именем A и массив 721 * 1440 с именем B, мне нужно найти коэффициенты корреляции Пирсона между двумя массивами, чтобы массив A, равный 4 * 4, коррелировал с каждой частью 4 * 4из массива B, форма которого 721 * 1440. Мой код Python приведен ниже, я новичок в Python и код 

import numpy as np
import pandas as pd
from scipy.stats.stats import pearsonr
A=np.arange(16).reshape(4,4)
B=np.arange(1038240).reshape(721,1440)

A1=np.asarray(A)
B1=np.asarray(B)
X=pd.DataFrame(A1)
Z1=pd.DataFrame(B1)
R=X.apply(lambda s: Z1.corrwith(s))

R
        0    1    2    3
0     1.0  1.0  1.0  1.0
1     1.0  1.0  1.0  1.0
2     1.0  1.0  1.0  1.0
3     1.0  1.0  1.0  1.0
4     1.0  1.0  1.0  1.0
...   ...  ...  ...  ...
1435  1.0  1.0  1.0  1.0
1436  1.0  1.0  1.0  1.0
1437  1.0  1.0  1.0  1.0
1438  1.0  1.0  1.0  1.0
1439  1.0  1.0  1.0  1.0

[1440 rows x 4 columns]
...