推荐系统数据格式COO Matrix

coo_matrix 是一种稀疏矩阵格式，代表坐标形式（Coordinate format）。在这种格式中，矩阵的非零元素通过行坐标和列坐标存储，因此适用于存储稀疏矩阵（即大部分元素为零的矩阵）。这种格式对于构建稀疏矩阵非常有效，特别是当你知道非零元素的具体位置时。

COO Matrix 解释

coo_matrix 的构造函数通常接受三个参数：

data：非零元素的值。
(row, col)：非零元素的行和列索引。
shape：矩阵的形状（行数和列数）。

代码分析

mat = coo_matrix((np.ones(len(user_ids)), (user_ids, item_ids)), shape=[n_user, n_item])

这段代码创建了一个稀疏矩阵，以下是详细解释：

np.ones(len(user_ids))：
- 生成一个长度为 len(user_ids) 的数组，所有元素都是 1。这表示我们假设每个用户-项目对的值都是 1。
(user_ids, item_ids)：
- user_ids 和 item_ids 是两个数组，分别表示用户和项目的索引。
- 例如，如果 user_ids = [0, 1, 2] 和 item_ids = [2, 0, 1]，这表示用户 0 对项目 2 有一个非零元素，用户 1 对项目 0 有一个非零元素，用户 2 对项目 1 有一个非零元素。
shape=[n_user, n_item]：
- 定义矩阵的形状，n_user 是用户的总数，n_item 是项目的总数。

示例

假设我们有以下数据：

user_ids = [0, 1, 2]
item_ids = [2, 0, 1]
n_user = 3
n_item = 3

这段代码会创建如下稀疏矩阵：

mat = coo_matrix((np.ones(3), ([0, 1, 2], [2, 0, 1])), shape=[3, 3])

生成的矩阵将是：

[[0 0 1][1 0 0][0 1 0]]

解释

(0, 2) 位置的值是 1，因为用户 0 对项目 2 有一个非零值。
(1, 0) 位置的值是 1，因为用户 1 对项目 0 有一个非零值。
(2, 1) 位置的值是 1，因为用户 2 对项目 1 有一个非零值。

完整示例代码

import numpy as np
from scipy.sparse import coo_matrix# 定义用户和项目的索引
user_ids = np.array([0, 1, 2])
item_ids = np.array([2, 0, 1])
n_user = 3
n_item = 3# 创建稀疏矩阵
mat = coo_matrix((np.ones(len(user_ids)), (user_ids, item_ids)), shape=[n_user, n_item])# 打印矩阵
print(mat.toarray())