Computer >> 컴퓨터 >  >> 프로그래밍 >> MySQL

MySQL 단일 쿼리에서 COUNT()와 IF()를 함께 사용하는 방법

데이터를 분석하다 보면 하나의 쿼리 안에서 특정 조건을 만족하는 행의 개수만 집계해야 하는 경우가 자주 발생합니다. MySQL에서는 COUNT() 함수와 IF() 함수를 조합하면 이런 작업을 아주 간단하게 처리할 수 있습니다. 핵심 원리는 COUNT() 함수가 NULL 값은 제외하고 계산한다는 점입니다.

1. 테이블 생성하기

먼저 예제에 사용할 테이블을 생성합니다.

mysql> create table DemoTable
(
    isValidUser boolean
);
Query OK, 0 rows affected (0.70 sec)

2. 샘플 데이터 삽입하기

INSERT 명령으로 테이블에 레코드를 여러 개 삽입해 보겠습니다.

mysql> insert into DemoTable values(true);
Query OK, 1 row affected (0.19 sec)
mysql> insert into DemoTable values(false);
Query OK, 1 row affected (0.07 sec)
mysql> insert into DemoTable values(false);
Query OK, 1 row affected (0.10 sec)
mysql> insert into DemoTable values(true);
Query OK, 1 row affected (0.11 sec)
mysql> insert into DemoTable values(true);
Query OK, 1 row affected (0.13 sec)
mysql> insert into DemoTable values(true);
Query OK, 1 row affected (0.28 sec)
mysql> insert into DemoTable values(false);
Query OK, 1 row affected (0.09 sec)

3. 저장된 데이터 확인하기

SELECT 문으로 테이블의 모든 레코드를 조회합니다.

mysql> select *from DemoTable;

실행 결과는 다음과 같습니다.

+-------------+
| isValidUser |
+-------------+
|           1 |
|           0 |
|           0 |
|           1 |
|           1 |
|           1 |
|           0 |
+-------------+
7 rows in set (0.00 sec)

boolean 타입은 내부적으로 true는 1, false는 0으로 저장되는 것을 확인할 수 있습니다.

4. COUNT()와 IF()를 조합한 조건부 카운트

이제 isValidUser 값이 1(true)인 행의 개수만 세는 쿼리를 살펴보겠습니다.

mysql> select count(if(isValidUser=1,1,NULL)) AS TOTAL_NUMBER_OF_ONE from DemoTable;

실행 결과:

+---------------------+
| TOTAL_NUMBER_OF_ONE |
+---------------------+
|                   4 |
+---------------------+
1 row in set (0.06 sec)

동작 원리 정리

IF(isValidUser=1, 1, NULL) 표현식은 조건이 참이면 1을 반환하고, 거짓이면 NULL을 반환합니다. 그리고 COUNT() 함수는 NULL이 아닌 값만 세기 때문에, 결과적으로 isValidUser가 1인 행의 개수만 집계됩니다. 전체 7개 행 중 true가 4개이므로 결과값은 4가 출력됩니다.

참고: SUM()을 활용한 대안

같은 결과는 SUM() 함수로도 얻을 수 있습니다. SELECT SUM(isValidUser=1) FROM DemoTable;처럼 작성하면 조건이 참인 경우 1, 거짓인 경우 0을 더해 동일한 집계 결과를 얻습니다. boolean 컬럼이라면 SUM(isValidUser)처럼 더 간결하게 표현할 수도 있습니다. 상황에 맞게 두 방식을 선택해 사용하시면 됩니다.