基於行總和的子集列

Question

我想基於行的總和來划分df列。

df數據框：

NE001  NE002  NE003  NE004
  2      0      0      2

我的預期輸出dfo ：

NE001  NE004
 2       2

我已經嘗試過dfo <- df[,which(names(df) == colSums(df==2))]但是不起作用。

有什么想法嗎？

Answer 1

您是否只是想將總計為2的列子集化？ 如果是這樣，您就很親密。 這是一個帶有較小數據幀的示例。

(d <- data.frame(x = c(2, 0), y = c(1, 2), z = c(1, 1)))
#   x y z
# 1 2 1 1
# 2 0 2 1

由於colSums(d) == 2返回邏輯向量，因此我們可以對其進行子集處理，並且將返回TRUE的列

colSums(d) == 2
#    x     y     z 
# TRUE FALSE  TRUE 
d[colSums(d) == 2]
#   x z
# 1 2 1
# 2 0 1