如何编写生成动物行为分析初级转移矩阵的函数？

核心实现步骤 #

先处理输入数据，通过配对连续姿态、统计频次来生成转移矩阵：

1. 提取姿态序列
   从输入数据框中取出单列的姿态向量：

df <- data.frame(c("A", "B", "A", "B", "C", "B", "C", "A", "C", "B", "A", "B"))
poses <- df[[1]]

2. 生成连续姿态对
   将每个姿态与它的下一个姿态配对，形成"前姿态-后姿态"的组合：

transition_pairs <- cbind(prev = poses[-length(poses)], next = poses[-1])

3. 统计转移频次并转矩阵
   用table()统计各类转移的次数，再转换为矩阵格式：

transition_matrix <- as.matrix(table(transition_pairs))

4. 标准化矩阵结构（可选）
   如果需要确保行列严格按指定顺序排列（比如A、B、C），并填充缺失的转移次数为0：

pose_levels <- c("A", "B", "C")
transition_matrix <- transition_matrix[pose_levels, pose_levels]
transition_matrix[is.na(transition_matrix)] <- 0

运行后得到的结果如下（注：原始数据中B→C仅出现2次，与你提供的期望矩阵存在差异，可检查原始数据是否有输入误差）：

next
prev A B C
  A 0 3 1
  B 2 0 2
  C 1 2 0

封装为可复用函数 #

将上述逻辑封装成函数，方便处理其他姿态序列：

generate_transition_matrix <- function(pose_vector) {
  # 生成连续姿态对
  transition_pairs <- cbind(prev = pose_vector[-length(pose_vector)], 
                            next = pose_vector[-1])
  # 统计频次并转换为矩阵
  trans_matrix <- as.matrix(table(transition_pairs))
  # 获取所有唯一姿态并排序
  pose_levels <- sort(unique(pose_vector))
  # 调整行列顺序，填充缺失值为0
  trans_matrix <- trans_matrix[pose_levels, pose_levels]
  trans_matrix[is.na(trans_matrix)] <- 0
  return(trans_matrix)
}

# 调用示例
generate_transition_matrix(df[[1]])

关键说明 #

• 矩阵的行代表前一姿态，列代表后一姿态，单元格数值为对应转移的发生频次
• 函数会自动识别输入向量中的所有姿态类别，无需手动指定
• 若数据中存在"姿态自身转移"的情况（比如连续出现两个A），函数会自动统计对应频次

内容的提问来源于stack exchange，提问作者Sissel