從數據幀中創建稀疏矩陣

Question

我試圖從數據框中創建稀疏數據矩陣，而不必構建導致嚴重內存問題的密集矩陣。

我發現了一個SO下面的帖子裏的解決方案似乎可以找到： Create Sparse Matrix from a data frame

我試過這個解決方案，但它不爲我工作，也許是因爲我的UserID和MovieID沒有按」 T T [開始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) 
trainingData 

UIMatrix <- sparseMatrix(i = trainingData$UserID, 
         j = trainingData$MovieID, 
         x = trainingData$Rating) 

dim(UIMatrix) 

我希望得到一個2×3矩陣，但變暗相當於最大用戶和電影的ID。

我已經嘗試過在帖子中建議的第二個解決方案，但它不會與可能的數據一起工作。

任何人都可以給一些建議嗎？

Answer 1

您可以將您的指數轉換爲從as.integer(as.factor(.))開始的指數。

UIMatrix <- sparseMatrix(i = as.integer(as.factor(trainingData$UserID)), 
         j = as.integer(as.factor(trainingData$MovieID)), 
         x = trainingData$Rating) 

dim(UIMatrix) 
# [1] 2 4 

dimnames(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