def sp_add2_array(sp,A):
# sp is a sparse matrix p-by-1
# A is an array of len p
# this function return an array corresponding to A+sp
ind = sp.nonzero()[0]
for i in ind:
A[i] += sp[i,0]
return A
def sp_add2_array(sp, arr):
"""
:param sp: sparse matrix p-by-1 (created
with :py:meth:`~rlpy.Tools.GeneralTools.sp_matrix`)
:param arr: a 1-D iterable type (ndarray, list, matrix) of length p.
Returns ret = arr + sp (with type(ret) = type(arr))
"""
ind = sp.nonzero()[0]
for i in ind:
arr[i] += sp[i, 0]
return arr
def sp_add2_array(sp, arr):
"""
:param sp: sparse matrix p-by-1 (created
with :py:meth:`~rlpy.Tools.GeneralTools.sp_matrix`)
:param arr: a 1-D iterable type (ndarray, list, matrix) of length p.
Returns ret = arr + sp (with type(ret) = type(arr))
"""
ind = sp.nonzero()[0]
for i in ind:
arr[i] += sp[i, 0]
return arr
