计算R中多列中的特定字

我有一个这样的数据框

df <- data.frame(id=c(1, 2, 3, 4, 5), staple_1=c("potato", "cassava","rice","fruit","coffee"), staple_2=c("cassava","beer","peanuts","rice","yams"), staple_3=c("rice","peanuts","fruit","fruit","rice"))

staples<-c("potato","cassava","rice","yams")

df$staples<-c(3,1,1,1,2)

---

df一个简单的
apply
就可以了

apply(df, 1, function(x) sum(staples %in% x))
#[1] 3 1 1 1 2

df$staples <- apply(df, 1, function(x) sum(staples %in% x))

apply（df，1，函数（x）和（订书钉%x中的%x））
#[1] 3 1 1 1 2
df$staples另一种方法是在列中循环，使用%
中的
%创建逻辑
向量的
列表
，并使用
Reduce
添加（
+
）

Reduce(`+`, lapply(df[-1], `%in%`, staples))
#[1] 3 1 1 1 2


或者使用
行和
将数据集转换为
矩阵
（不带'id'列），使用%
中的
%转换为逻辑
向量
，分配
dim
转换为与
df[-1]
维度相同的维度，并获得
行和

rowSums(`dim<-`(as.matrix(df[-1]) %in% staples, dim(df[-1])))
#[1] 3 1 1 1 2

行和（`dim这里是一个整洁的解决方案

library(tidyverse)

df <- data_frame(id = c(1, 2, 3, 4, 5), 
             staple_1 = c("potato", "cassava", "rice", "fruit", "coffee"), 
             staple_2 = c("cassava", "beer", "peanuts", "rice", "yams"), 
             staple_3 = c("rice", "peanuts", "fruit", "fruit", "rice"))

staples_vect <- c("potato", "cassava", "rice", "yams")

df %>% 
     mutate(staples = pmap_int(select(., starts_with("staple_")), ~sum(c(...) %in% staples_vect)))

库（tidyverse）
请张贴你的方法