最有效的方法来创建非冗余相关矩阵Python?

时间:2016-02-24 20:40:40

标签: python pandas matrix dataframe adjacency-matrix

我觉得像numpy,scipy或者networkx有办法做到这一点,但我还没想到它。

我的问题是如何以最有效的方式(在Python中)从大型数据集的冗余相关矩阵中以数据帧形式创建非冗余相关矩阵?

我在7000x7000矩阵上使用这种方法,并且它永远占用我的MacBook Air 4GB Ram(我知道,我绝对不应该将它用于编程,但是那个' s另一个讨论)

冗余相关矩阵的示例

enter image description here

非冗余相关矩阵的示例

enter image description here

我在下面做了一个非常天真的方式,但必须有一个更好的方法。我喜欢将我的矩阵存储在稀疏矩阵中并将它们转换为数据帧以用于存储目的。

import pandas as pd
import numpy as np
import networkx as nx

#Example DataFrame
L_test = [[0.999999999999999,
  0.374449352805868,
  0.000347439531148995,
  0.00103026903356954,
  0.0011830950375467401],
 [0.374449352805868,
  1.0,
  1.17392596672424e-05,
  1.49428208843456e-07,
  1.216664263989e-06],
 [0.000347439531148995,
  1.17392596672424e-05,
  1.0,
  0.17452569907144502,
  0.238497202355299],
 [0.00103026903356954,
  1.49428208843456e-07,
  0.17452569907144502,
  1.0,
  0.7557000865939779],
 [0.0011830950375467401,
  1.216664263989e-06,
  0.238497202355299,
  0.7557000865939779,
  1.0]]
labels = ['AF001', 'AF002', 'AF003', 'AF004', 'AF005']
DF_1 = pd.DataFrame(L_test,columns=labels,index=labels)

#Create Nonredundant Similarity Matrix
n,m = DF_test.shape #they will be the same since it's adjacency
#Empty array to fill
A_tmp = np.zeros((n,m)) 
#Copy part of the array
for i in range(n):
    for j in range(m):
        A_tmp[i,j] = DF_test.iloc[i,j]
        if j==i:
            break
#Make array sparse for storage
A_csr = csr_matrix(A_tmp) 
#Recreate DataFrame
DF_2 = pd.DataFrame(A_csr.todense(),columns=DF_test.columns,index=DF_test.index) 
DF_2.head()

1 个答案:

答案 0 :(得分:2)

我认为您可以使用np.tril创建数组,然后使用DataFrame DF_1多个数组:

print np.tril(np.ones(DF_1.shape))
[[ 1.  0.  0.  0.  0.]
 [ 1.  1.  0.  0.  0.]
 [ 1.  1.  1.  0.  0.]
 [ 1.  1.  1.  1.  0.]
 [ 1.  1.  1.  1.  1.]]

print np.tril(np.ones(DF_1.shape)) * DF_1
          AF001         AF002     AF003   AF004  AF005
AF001  1.000000  0.000000e+00  0.000000  0.0000      0
AF002  0.374449  1.000000e+00  0.000000  0.0000      0
AF003  0.000347  1.173926e-05  1.000000  0.0000      0
AF004  0.001030  1.494282e-07  0.174526  1.0000      0
AF005  0.001183  1.216664e-06  0.238497  0.7557      1