2

我想编写一个将数据框转换为矩阵的函数。数据框是事件列表。每行对应一个访问或购买产品的人。

my.df <- data.frame(person = c('A', 'A', 'B', 'B', 'B', 'C'),
                    week = c(1, 2, 1, 3, 3, 2),
                    event = c('visit', 'buy', 'visit', 'visit', 'buy', 'visit'))
> my.df
  person week event
1      A    1 visit
2      A    2   buy
3      B    1 visit
4      B    3 visit
5      B    3   buy
6      C    2 visit

所需的输出矩阵将行作为人，将列作为周。在 (person, week) 条目中，如果有人购买，我希望有“购买”，如果没有，如果有人访问，我希望有“访问”，否则我希望有“无”作为条目。更具体地说，所需的输出是以下矩阵：

> my.mat
  1       2      3      
A "visit" "buy"  "none" 
B "visit" "none" "buy"  
C "none"  "none" "visit"

我的想法是我应该将事件转换为数字，使用最大值进行强制转换，然后将数字转换回事件，但我不完全确定如何将它们放在一起。

4

3 回答 3

2

正如 Arun 指出的，使用这个reshape2包：

library(reshape2)

# there is a variety of ways to get the precedence you like
# I chose to just sort the strings
acast(my.df, person ~ week, function(x) {sort(as.character(x))[1]},
      value.var = 'event', fill = 'none')
#  1       2       3     
#A "visit" "buy"   "none"
#B "visit" "none"  "buy" 
#C "none"  "visit" "none"
于 2013-08-22T21:24:02.847 回答
1

只是一段代码：

unique(event)
as.numeric(factor(event))
unique(event)[as.numeric(factor(event)[1])]

第一行显示你有多少不同的事件。第二个将您的事件转换为数字。第三个将给出与编号元素相关的文本（此处为 1）。

于 2013-08-22T22:17:13.443 回答
1

在@eddi 和@Rodrigo 的答案的基础上，我设法找到了以下有点冗长但有效的代码。如果我想要更复杂的事件排序，它也可以工作。

require(reshape2) # For acast(...)

# Input data frame
my.df <- data.frame(person = c('A', 'A', 'B', 'B', 'B', 'C'),
                    week = c(1, 2, 1, 3, 3, 2),
                    event = c('visit', 'buy', 'visit', 'visit', 'buy', 'visit'))

# Convert event into numbers, with buy > visit
the.levels <- c('visit', 'buy')
my.df$event <- as.numeric(factor(my.df$event, levels = the.levels))

# Build matrix
temp <- acast(my.df, person ~ week, function(x) {max(x)},
             value.var = 'event', fill = 0)

# Convert event numbers back into text
number.to.event <- as.list(setNames(c('none', 'visit', 'buy'),
                                as.character(c(0, 1, 2))))
# Save row names and column names
out <- matrix(number.to.event[as.character(temp)], nrow = 3,
              dimnames = dimnames(temp))

> out
  1       2       3     
A "visit" "buy"   "none"
B "visit" "none"  "buy" 
C "none"  "visit" "none"
于 2013-08-23T01:12:11.617 回答