我正在关注这个stackoverflow post on csr matrix multiplication to a vector并在python中实现它并获得列表超出范围错误。
这是我的代码:
def MatrixMultiplication(data,row_ptr,col_ptr,vec):
ResultMatrix =[]
vec_len = len(vec)
for i in range(0,vec_len):
ResultMatrix.insert(i,0)
for i in range(0,vec_len):
start, end = row_ptr[i], row_ptr[i + 1]
for k in range(start, end):
ResultMatrix[i] = ResultMatrix[i]+data[k]*vec[col_ptr[k]]
return ResultMatrix
data = [2, 4, 7, 1, 3, 2]
row_ptr = [2,3 ,5, 5 ,6]
col_ptr = [1 ,3, 4, 0, 3, 3]
vec = [2,3, 5, 4, 2]
MatrixMultiplication(data,row_ptr,col_ptr,vec)
请帮我解决我出错的地方。
输出应为:[22 14 14 0 8]
错误:
IndexError: list index out of range
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<command-338158343473691> in <module>()
----> 1 MatrixMultiplication(data,row_ptr,col_ptr,vec)
<command-3658506804172571> in MatrixMultiplication(data, row_ptr, col_ptr, vec)
5 ResultMatrix.insert(i,0)
6 for i in range(0,vec_len):
----> 7 start, end = row_ptr[i], row_ptr[i + 1]
8 for k in range(start, end):
9 ResultMatrix[i] = ResultMatrix[i]+data[k]*vec[col_ptr[k]]
IndexError: list index out of range
供参考:
row_ptr的最后一个元素是数据列表的大小
错误消息非常明显:您试图在for循环中访问row_ptr[i + 1]
,该循环最多为vec_len
,这是列表的长度。当你到达for循环和i = vec_len - 1
的最后一次迭代,然后i + 1 = vec_len
,这超出了你的列表范围(记住,Python列表是0初始化的)。
为防止出现此错误,您的范围应仅在第二个for循环中达到vec_len - 1
。