Auswahl von Spalten der Matrix basierend auf binären Vektor

Question

In einem Stück meines Codes, muss ich bestimmte Spalten meiner Matrix basierend auf binären Matrix, die ich habe, auswählen, und speichern Sie es in der Liste, aber ich Gesicht mit folgendem Problem. Weiß jemand, was das Problem ist? Hier ist meine Matrix und Code:

> data 
     A B C D 
    [1,] 1 6 11 16 
    [2,] 2 7 12 17 
    [3,] 3 8 13 18 
    [4,] 4 9 14 19 
    [5,] 5 10 15 20 

    > select<-c(1,0,1,0) 

> p<-data[,select, drop=FALSE] 
> p 
    A A 
[1,] 1 1 
[2,] 2 2 
[3,] 3 3 
[4,] 4 4 
[5,] 5 5 

Meine erwartete Ausgabe ist:

> p 
    A C 
[1,] 1 11 
[2,] 2 12 
[3,] 3 13 
[4,] 4 14 
[5,] 5 15 

Answer 1

Sie werden es zu einem logischen Vektor konvertieren müssen oder sonst wird es 1 und 0 als Spaltennummern behandeln:

data[,as.logical(select), drop=F] 
# A C 
# 1 1 11 
# 2 2 12 
# 3 3 13 
# 4 4 14 
# 5 5 15 

Answer 2

Sie können es auch versuchen. Wenn Sie nur select liefern, geben Sie an, Spalten nach Spaltennummer auszuwählen. Wenn Sie select==1 liefern, erhalten Sie zuerst einen logischen Vektor und verwenden diesen, um Spalten auszuwählen.

data <- matrix(1:20, nrow = 5, dimnames = list(NULL, c("A", "B", "C", "D"))) 
select <- c(1, 0, 1, 0) 
data[, select == 1] 
##  A C 
## [1,] 1 11 
## [2,] 2 12 
## [3,] 3 13 
## [4,] 4 14 
## [5,] 5 15