[R기초] 조건에 맞는 원소의 위치를 반환 (which 함수)

[R기초] 조건에 맞는 원소의 위치를 반환 (which 함수)

1. 벡터

벡터와 비교연산자를 이용하여 조건을 대입하면, 해당 조건에 만족하는 원소의 위치를 반환합니다. 

> a=c(1,2,3,1,2,3)

> which(a==3)

[1] 3 6

> which(a>1)

[1] 2 3 5 6

2. 데이터프레임

데이터프레임도 벡터와 원리는 같습니다. arr.ind 옵션을 TRUE로 설정하면 해당 원소의 행의 위치와 열의 위치를 각각 반환합니다. 

> df=data.frame(c1=c(1,2,3),c2=c(1,1,3),c3=c(1,2,2))

> df

  c1 c2 c3

1  1  1  1

2  2  1  2

3  3  3  2

> which(df==1)

[1] 1 4 5 7

> which(df==1,arr.ind=TRUE)

     row col

[1,]   1   1

[2,]   1   2

[3,]   2   2

[4,]   1   3