我希望您能在Python中(假定使用numpy和/或scipy)计算一个向量与数组的每一行之间的Pearson相关系数。由于真实数据数组的大小和内存限制,无法使用标准相关矩阵计算函数。以下是我的天真实现:
任何加速此过程的想法都将不胜感激!
import numpy as np
import scipy.stats as sps
np.random.seed(0)
def correlateOneWithMany(one, many):
"""Return Pearson's correlation coef of 'one' with each row of 'many'."""
pr_arr = np.zeros((many.shape[0], 2), dtype=np.float64)
pr_arr[:] = np.nan
for row_num in np.arange(many.shape[0]):
pr_arr[row_num, :] = sps.pearsonr(one, many[row_num, :])
return pr_arr
obs, varz = 10 ** 3, 500
X = np.random.uniform(size=(obs, varz))
pr = correlateOneWithMany(X[0, :], X)
%timeit correlateOneWithMany(X[0, :], X)
# 10 loops, best of 3: 38.9 ms per loop
任何加速此过程的想法都将不胜感激!