2013-06-02 8 views
6

Ho due csr_matrix, uniFeature e biFeature.Come concatenare due matrici in Python?

Desidero una nuova matrice Feature = [uniFeature, biFeature]. Ma se li concateno direttamente in questo modo, c'è un errore che dice che la matrice Feature è una lista. Come posso ottenere la concatenazione matriciale e ottenere lo stesso tipo di matrice, ad esempio csr_matrix?

E non funziona se faccio questo dopo la concatenazione: Feature = csr_matrix(Feature) dà l'errore:

Traceback (most recent call last): 
    File "yelpfilter.py", line 91, in <module> 
    Feature = csr_matrix(Feature) 
    File "c:\python27\lib\site-packages\scipy\sparse\compressed.py", line 66, in __init__ 
    self._set_self(self.__class__(coo_matrix(arg1, dtype=dtype))) 
    File "c:\python27\lib\site-packages\scipy\sparse\coo.py", line 185, in __init__ 
    self.row, self.col = M.nonzero() 
TypeError: __nonzero__ should return bool or int, returned numpy.bool_ 

risposta

15

Il modulo scipy.sparse comprende le funzioni hstack e vstack.

Ad esempio:

In [44]: import scipy.sparse as sp 

In [45]: c1 = sp.csr_matrix([[0,0,1,0], 
    ...:      [2,0,0,0], 
    ...:      [0,0,0,0]]) 

In [46]: c2 = sp.csr_matrix([[0,3,4,0], 
    ...:      [0,0,0,5], 
    ...:      [6,7,0,8]]) 

In [47]: h = sp.hstack((c1, c2), format='csr') 

In [48]: h 
Out[48]: 
<3x8 sparse matrix of type '<type 'numpy.int64'>' 
    with 8 stored elements in Compressed Sparse Row format> 

In [49]: h.A 
Out[49]: 
array([[0, 0, 1, 0, 0, 3, 4, 0], 
     [2, 0, 0, 0, 0, 0, 0, 5], 
     [0, 0, 0, 0, 6, 7, 0, 8]]) 

In [50]: v = sp.vstack((c1, c2), format='csr') 

In [51]: v 
Out[51]: 
<6x4 sparse matrix of type '<type 'numpy.int64'>' 
    with 8 stored elements in Compressed Sparse Row format> 

In [52]: v.A 
Out[52]: 
array([[0, 0, 1, 0], 
     [2, 0, 0, 0], 
     [0, 0, 0, 0], 
     [0, 3, 4, 0], 
     [0, 0, 0, 5], 
     [6, 7, 0, 8]]) 
+0

Grazie mille! Esattamente quello che mi serve –

+0

Ho ricevuto questo errore: TypeError: vstack() ha ottenuto un argomento di parole chiave inaspettate 'format' – Moh

+0

RISOLTO: Il problema era: Invece di importare il modulo scipy.parse, avevo importato scipy – Moh