R语言输出满足条件的元素的值，急切求助！！

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

如下表，列分别表示用户ID， 电影ID和评分ID

	UserID	MovieID	Rating
1	43	23	3
2	43	42	1
3	2	123	2
4	58	3	5
5	2	545	4
6	44	8	3
7	3	45	2
8	2	32	1
9	43	44	4

如何得到每个用户其评分大于3的全部电影，并且将这些电影按照ID的升序作为一行写入到文本文档中。
例如：
User1:3,4,6,7
User2： 45,123,545   
...............
User43： 23,42,44
感谢各位大神了！

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：R语言 RATING UserID Movie User 查询 输出

movie <- data.frame(list(UserID=c(43,43,2,58),MovieID=c(23,42,123,3),Rating=c(3,1,2,5)))
movie <- subset(movie,Rating>=3)
movie <- movie[with(movie,order(MovieID)),]
write.table(movie, file = "movie.txt")
自己参考改一下吧

其实，我比照着写了一下代码。。。不行咯

1. 初学者来献丑给个答案，没理解你是需要按照 UserID 排序还是按照 UserID & MovieID 一起排序，我两个都放进去了，data 数据我从网页复制后随便改了几个数字。
   
2. 
   
3. data
   
4.   UserID MovieID Rating
   
5. 1     43      23      4
   
6. 2     43      42      3
   
7. 3      2     123      2
   
8. 4     58       3      5
   
9. 5      2     545      4
   
10. 6     44       8      3
    
11. 7      3      45      2
    
12. 8      2      32      3
    
13. 9     43      44      4
    
14. data1 <- data[which(data$Rating >= 3),]
    
15. data2 <- data1[order(data1$UserID,data1$MovieID),]
    
16. data3 <- aggregate(data = data2,MovieID~UserID,FUN = function(x) list(x))
    
17. 
    
18. data3
    
19.   UserID    MovieID
    
20. 1      2    32, 545
    
21. 2     43 23, 42, 44
    
22. 3     44          8
    
23. 4     58          3
    
24. 
    
25. 写入TXT的过程没写，这个过程你自己琢磨下吧，write.table 即可实现。

复制代码

咦，排序那一步，有几个 美元 符号不见了，不过大家应该都看得懂。。。。

这个问题也可以用dplyr包来实现，几个关键的函数是group_by, filter和arrange。在这里我用%>%符使程序更简洁一些(可读性未必好):

1. movie <- data.frame(UserID = c(43, 43, 2, 58, 2, 44, 3, 2, 43),
   
2.                     MovieID = c(23, 42, 123, 3, 545, 8, 45, 32, 44),
   
3.                     Rating = c(3, 1, 2, 5, 4, 3, 2, 1, 4))
   
4. 
   
5. movie
   
6. 
   
7. library(dplyr)
   
8. 
   
9. movie2 <- movie %>%
   
10.   group_by(UserID) %>%
    
11.   filter(Rating >= 3) %>%
    
12.   arrange(UserID)

复制代码

不知为何原回帖无法编辑，所以重新回复。相对于整理数据，这个问题更困难的地方在于写文件---因为MovieID每个元素的长度不同， 写文件决非直接使用write.table那么简单。下面是我的一种解法尝试：用循环，每次循环写入一行数据。

1. movie <- data.frame(UserID = c(43, 43, 2, 58, 2, 44, 3, 2, 43),
   
2.                     MovieID = c(23, 42, 123, 3, 545, 8, 45, 32, 44),
   
3.                     Rating = c(3, 1, 2, 5, 4, 3, 2, 1, 4))
   
4. library(dplyr)
   
5. 
   
6. movie1 <- movie %>%
   
7.   group_by(UserID) %>%
   
8.   filter(Rating >= 3) %>%
   
9.   arrange(UserID)
   
10. 
    
11. movie2 <- aggregate(MovieID ~ UserID, data = movie1[, -3], list)
    
12. 
    
13. # Do some preparation for writing .txt
    
14. UserID_new <- paste0("User", movie2$UserID, ":")
    
15. movieid <- movie2$MovieID # It is a list!
    
16. 
    
17. # Write the file
    
18. 
    
19. for (i in 1:nrow(movie2)) {
    
20.   cat(UserID_new[i], movieid[[i]], "\n", file = "output.txt", append = TRUE)
    
21. }

复制代码

（不得不吐槽一下，论坛的回帖功能实在太不稳定了，每次编辑都会删掉我贴出代码的一部分，令人十分郁闷。什么时候能修复这方面的bug?）

止水之涯 发表于 2016-11-14 22:02

“用write.table即可实现”是不对的。您可以试一试，会报错。

hugebear 发表于 2016-11-15 14:59
“用write.table即可实现”是不对的。您可以试一试，会报错。

是的，是会报错，学习了~~~

hugebear 发表于 2016-11-15 14:45
不知为何原回帖无法编辑，所以重新回复。相对于整理数据，这个问题更困难的地方在于写文件---因为MovieID每 ...

（不得不吐槽一下，论坛的回帖功能实在太不稳定了，每次编辑都会删掉我贴出代码的一部分，令人十分郁闷。什么时候能修复这方面的bug?）

不知是哪一部分，我这里没有测试出来