Skip to content

Instantly share code, notes, and snippets.

@viveksck

viveksck/gist:64bfb3fbd7937bd11778705f94140e34

Forked from seanjtaylor/gist:568141f04a16d518be24
Created March 17, 2019 01:22
Show Gist options
  • Save viveksck/64bfb3fbd7937bd11778705f94140e34 to your computer and use it in GitHub Desktop.
Save viveksck/64bfb3fbd7937bd11778705f94140e34 to your computer and use it in GitHub Desktop.
Download ZIP
Reshaping a Pandas dataframe into a sparse matrix
Raw
gistfile1.py
import pandas as pd
import scipy.sparse as sps
df = pd.DataFrame({'tag1': ['sean', 'udi', 'bogdan'], 'tag2': ['sean', 'udi', 'udi'], 'freq': [1,2,3]})
# tag1 -> rows, tag2 -> columns
df.set_index(['tag1', 'tag2'], inplace=True)
mat = sps.coo_matrix((df.freq, (df.index.labels[0], df.index.labels[1])))
print(mat.todense())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment