how to merge two csr_matrix, example python source code

let's see the code.

..

from scipy.sparse import csr_matrix

import numpy as np

#first matrix

row = np.array([0, 0, 1, 2, 2, 2])

col = np.array([0, 2, 2, 0, 1, 2])

data = np.array([1, 1, 1, 1, 1, 1])

mtx = csr_matrix((data, (row, col)), shape=(3, 3))

#second matrix

row = np.array([0, 0, 1, 2, 2, 2])

col = np.array([0, 1, 2, 0, 1, 2])

data = np.array([1, 1, 1, 1, 1, 1])

mtx2 = csr_matrix((data, (row, col)), shape=(3, 3))

#merge two matrix

mtx3 = merge_two_csr_mtx(mtx, mtx2)

#check

print('1st\n',mtx)

print('2nd\n',mtx2)

print('merge\n',mtx3)

..

result

1st
   (0, 0) 1
  (0, 2) 1
  (1, 2) 1
  (2, 0) 1
  (2, 1) 1
  (2, 2) 1
2nd
   (0, 0) 1
  (0, 1) 1
  (1, 2) 1
  (2, 0) 1
  (2, 1) 1
  (2, 2) 1
merge
   (0, 0) 2.0
  (0, 1) 1.0
  (0, 2) 1.0
  (1, 2) 2.0
  (2, 0) 2.0
  (2, 1) 2.0
  (2, 2) 2.0