Pandas pivot to sparse

from scipy.sparse import csr_matrix
from pandas.api.types import CategoricalDtype

person_c = CategoricalDtype(sorted(frame.person.unique()), ordered=True)
thing_c = CategoricalDtype(sorted(frame.thing.unique()), ordered=True)

row = frame.person.astype(person_c).cat.codes
col = frame.thing.astype(thing_c).cat.codes
sparse_matrix = csr_matrix((frame["count"], (row, col)), \
                           shape=(person_c.categories.size, thing_c.categories.size))

>>> sparse_matrix
<3x4 sparse matrix of type '<class 'numpy.int64'>'
     with 6 stored elements in Compressed Sparse Row format>

>>> sparse_matrix.todense()
matrix([[0, 1, 0, 1],
        [1, 0, 0, 1],
        [1, 0, 1, 0]], dtype=int64)


dfs = pd.SparseDataFrame(sparse_matrix, \
                         index=person_c.categories, \
                         columns=thing_c.categories, \
                         default_fill_value=0)
>>> dfs
        a   b   c   d
 him    0   1   0   1
  me    1   0   0   1
 you    1   0   1   0

Source: stackoverflow.com

Similar codes

1 vote106 views1 answer
Def dft(self, array): stack = Stack() visited = set() stack.append(self) while len(stack) > 0 and while len(visited) >= 0: current = stack.pop() array.append(current) visited.add(current) return array
Python
8 votes100 views2 answers
How to check whether input is string or not
Python
0 vote129 views1 answer
Candle stick with volume plotly
Python
1 vote106 views1 answer
Why do we use graphs?
Python
1 vote105 views1 answer
Discord.py owner only commands
Python
0 vote99 views1 answer
Django change foreign key
Python
6 votes107 views1 answer
Write a Python program to append text to a file and display the text.
Python
2 votes106 views1 answer
Selenium keep window open python
Python
0 vote95 views1 answer
Convert pandas group to dict
Python
1 vote109 views1 answer
Module turtle has no forward member
Python