我试图从数据框创建一个稀疏数据矩阵，避免构建一个会导致严重内存问题的密集矩阵。

我在Stack Overflow上找到了一个似乎有解决方案的帖子：从数据框创建稀疏矩阵

我尝试了这个解决方案，但对我不起作用，可能是因为我的UserID和MovieID不是从1开始的。

这是我的示例代码：

library(Matrix)UserID<-c(10090,10090,10090,10316,10316)MovieID <-c(63155,63530,63544,63155,63545)Rating <-c(2,2,1,2,1)trainingData<-data.frame(UserID,MovieID,Rating)trainingDataUIMatrix <- sparseMatrix(i = trainingData$UserID,                         j = trainingData$MovieID,                         x = trainingData$Rating)dim(UIMatrix)

我期望得到一个2×3的矩阵，但维度对应的是最大用户和电影ID。

我尝试了帖子中建议的第二个解决方案，但它对我的数据也不起作用。

谁能给些建议？

回答：

你可以使用as.integer(as.factor(.))将你的索引转换为从1开始的索引。

UIMatrix <- sparseMatrix(i = as.integer(as.factor(trainingData$UserID)),                         j = as.integer(as.factor(trainingData$MovieID)),                         x = trainingData$Rating)dim(UIMatrix)# [1] 2 4dimnames(UIMatrix) <- list(sort(unique(trainingData$UserID)),                           sort(unique(trainingData$MovieID)))UIMatrix# 2 x 4 sparse Matrix of class "dgCMatrix"#       63155 63530 63544 63545# 10090     2     2     1     .# 10316     2     .     .     1