SQL로 For Loop(For문/반복작업) 구현하기

1.왜 이게 필요했을까?

  • 여러 언어를 오고가면서 분석작업을 진행할 때가 많은데,
  • 가끔은 SQL로만 하고 싶을 때가 있고, 그 때마다 문제가 되었던 부분 중 하나가 For Loop였다. 그래서 찾아보다가 알게 되었다.

2.구현하기

  • presto(0.241)에서 진행하였다.

A.기본 Sequence Table 생성

--Sequence Table 생성
CREATE TABLE temp.sequence AS (
    SELECT 0 AS num
    UNION ALL SELECT 1  AS num
    UNION ALL SELECT 2  AS num
    UNION ALL SELECT 3  AS num
    UNION ALL SELECT 4  AS num
    UNION ALL SELECT 5  AS num
    UNION ALL SELECT 6  AS num
    UNION ALL SELECT 7  AS num
    UNION ALL SELECT 8  AS num
    UNION ALL SELECT 9  AS num
)

Copy

B.필요한 반복수만큼의 Sequence 테이블 생성

-- 시퀀스 테이블의 조합을 통해서 For Loop을 하기 위한 Sequence 생성합니다.
--100번 돌리기 위한 방법

-- CREATE TABLE temp.for_loop AS (
--     SELECT 1 + s1.num + s2.num * 10 AS i
--     FROM temp.sequence s1
--     CROSS JOIN temp.sequence s2
--     ORDER BY i
-- )

-- 1000 번 돌리기 위한 방법
CREATE TABLE temp.for_loop AS (
    SELECT 1 + s1.num + s2.num * 10 + s3.num * 100 AS i
    FROM temp.sequence s1
    CROSS JOIN temp.sequence s2
    CROSS JOIN temp.sequence s3    
    ORDER BY i
)

Copy

C.사용하기

  • 이제 위 for_loop와 반복실행하려는 작업을 cross join으로 묶어 진행하면 된다.

3.예시( 동전 던지기 시뮬레이션)

A.확률 테이블 생성

CREATE TABLE temp.coin
(
      coin_name      varchar(10),
      probability    double
)

Copy

INSERT INTO temp.coin
VALUES ('FairCoin', 0.5),
       ('BiasCoin', 0.9)

Copy

B.For Loop 돌리기

CREATE TABLE temp.experiments AS (       
    SELECT c.coin_name, 
           CASE WHEN random() > c.probability THEN 'head' ELSE 'tail' END AS experiment
FROM temp.coin AS c
CROSS JOIN temp.for_loop AS f
)

Copy

C.결과 요약

SELECT coin_name, 
       COUNT(CASE WHEN experiment = 'head' then 1 end) as n_heads, 
       COUNT(*) as n_flips,  
       cast(COUNT(CASE WHEN experiment = 'head' then 1 end) as double) / COUNT(*) as head_prob
FROM temp.experiments
GROUP BY coin_name

Copy

4.결론

  • Cross Join을 사용하기 때문에 리소스를 엄청 먹기 때문에 주의를 요하지만, 가끔 쓰기에는 아주 좋은 방법이다. ?

5.Footnotes

Read more

[책]Reshuffle: Who wins when AI restacks the knowledge economy

[책]Reshuffle: Who wins when AI restacks the knowledge economy

원래는 Amazon에 가서 Personal Knowledge Managment에 관한 책을 사려고 했다. Sketch Your Mind라는 책이었는데, 그 때 이 책 “Reshuffle”을 발견하였다. AI가 어떻게 Knowledge Economy를 흔들 것가? 라는 부제를 훑어보면서 저자가 쓴 다른 책을 보게 되었는데 거기에 내가 좋아했던 책을쓴 저자라는 것을 알게 되었다. 그래서 크게 고민하지 않고 구매를 하고

By Bongho, Lee
[책]올라운드투자, 누군가의 투자일기

[책]올라운드투자, 누군가의 투자일기

“올라운드 투자”라는 제목을 보았을 때는, “올라운드 플레이어”가 생각이 났다. “올라운드”라는 표현을 오랜만에 들어본 까닭이었다. 그럼에도 불구하고 이 책을 고른 것은 저자가 그간 보여준 컨텐츠에 대한 신뢰가 있던 까닭이었다. 컨텐츠를 다양하게 보는 편이지만 깊이가 아주 있지는 않았다. 여기서 깊이라 함은 기존 전문적인 정량적 분석의 내용의 수준을 말하는 것이다.

By Bongho, Lee
내가 놓치고 있던 미래, 먼저 온 미래를 읽고

내가 놓치고 있던 미래, 먼저 온 미래를 읽고

장강명 작가의 책은, 유학시절 읽고 처음이었다. 유학시절 "한국이 싫어서"라는 책은 동기부여가 상당히 되는 책이었다. 한국을 떠나 새로운 정채성을 학생으로서 Build up 해나가고 있던 상황에서 이 책은 제목부터 꽤 솔깃하였다. 물론 결말이 기억날 정도로 인상깊은 책은 아니었지만 말이다. 그렇게 시간이 흘러 장강명 작가의 책은 더 이상 읽지 않던

By Bongho, Lee