Aggiungere matrice NumPy come colonna Pandas frame di dati

Question

Ho un oggetto Pandas trama di dati di forma (X, Y) che assomiglia a questo:

[[1, 2, 3], 
[4, 5, 6], 
[7, 8, 9]] 

e NumPy matrice sparsa (CSC) di forma (X , Z) che sembra qualcosa di simile

[[0, 1, 0], 
[0, 0, 1], 
[1, 0, 0]] 

Come posso aggiungere il contenuto dalla matrice al frame di dati in una nuova colonna denominata in modo che il frame di dati finirà in questo modo:

[[1, 2, 3, [0, 1, 0]], 
[4, 5, 6, [0, 0, 1]], 
[7, 8, 9, [1, 0, 0]]] 

Si noti che il frame di dati ora ha forma (X, Y + 1) e le righe dalla matrice sono elementi nel frame di dati.

Answer 1

import numpy as np 
import pandas as pd 
import scipy.sparse as sparse 

df = pd.DataFrame(np.arange(1,10).reshape(3,3)) 
arr = sparse.coo_matrix(([1,1,1], ([0,1,2], [1,2,0])), shape=(3,3)) 
df['newcol'] = arr.toarray().tolist() 
print(df) 

cede

0 1 2  newcol 
0 1 2 3 [0, 1, 0] 
1 4 5 6 [0, 0, 1] 
2 7 8 9 [1, 0, 0] 

Answer 2

considerare l'utilizzo di un datastructure dimensionale superiore (un Panel), piuttosto che la memorizzazione di un array in colonna:

In [11]: p = pd.Panel({'df': df, 'csc': csc}) 

In [12]: p.df 
Out[12]: 
    0 1 2 
0 1 2 3 
1 4 5 6 
2 7 8 9 

In [13]: p.csc 
Out[13]: 
    0 1 2 
0 0 1 0 
1 0 0 1 
2 1 0 0 

Guarda sezioni, ecc, ecc , ecc.

In [14]: p.xs(0) 
Out[14]: 
    csc df 
0 0 1 
1 1 2 
2 0 3 

See the docs for more on Panels.