Google Colab이 AI와 연동되어 매우 사용하기 편해진 이야기

취미로 주식 분석을 실시하고 있습니다. AI 시대이기 때문에, 코드를 직접 작성하는 것이 아니라 AI를 사용하여 표나 그림으로 시각화하고, 시사점을 얻어 2차 분석, 3차 분석을 실시하고 있습니다.

AI를 사용하여 분석할 때, VSCode를 일일이 실행하여 분석하는 것도 번거롭다고 생각했습니다. 그래서 오랜만에 Google Colab을 열어보았는데, 굉장히 다루기 쉬워져 있었습니다. 그래서 이번에는 Google Colab이 다루기 쉬워진 이야기에 대해 해보겠습니다.

Google Colab (Collaboratory)은 브라우저에서 직접 Python을 실행할 수 있는 호스트형 노트북 서비스입니다. 환경 구축이 전혀 필요 없이 사용할 수 있기 때문에 데이터 분석의 정석적인 도구로 널리 사용되고 있습니다.

이 Google Colab은 최근 업데이트를 통해 AI (Gemini)와의 융합이 극적으로 진행되었습니다.

2023년 5월: 미국에서 유료 플랜 사용자를 대상으로 코딩 지원 기능인 「Colab AI」로 발표. -
2023년 10월: 일본 국내 환경에서도 이용 가능해짐. -
2024년 6월: 일반 무료 사용자에게도 AI에 의한 코드 보완 및 채팅 기능이 폭넓게 개방. -
2025년 6월: 「AI 퍼스트 (AI-first)」 환경으로 쇄신. 또한 터미널 (Terminal) 기능도 모든 사용자에게 무료로 개방되었습니다.

현재는 화면 오른쪽에 Gemini 채팅 패널이 상주하고 있어, 데이터 취득 지시부터 시각화 제안까지 직관적인 일본어 프롬프트(Prompt)만으로 동행해 줍니다.

이번에는 최근의 지정학적 리스크(이란 정세의 긴박화)가 **「일본의 주요 3대 시장(Prime, Standard, Growth)에 어떤 영향을 주었는가」**를 테마로, Colab AI를 사용하여 검증해 보겠습니다.

통상적이라면 라이브러리 선정부터 정규화 계산식, 그래프 그리기 코딩까지 VSCode 등에서 열심히 작성해야 하지만, 새로워진 Google Colab의 AI를 사용하면 단 하나의 프롬프트로 완결됩니다.

Google Drive에서 Google Colab 파일 생성

Colab의 AI 채팅에 다음과 같이 자신이 하고 싶은 것을 그대로 일본어로 입력합니다.

프롬프트 예시:

최근의 지정학적 리스크(이란 정세 긴박화 등)에 따른 일본 시장에 대한 영향을 검증하고 싶습니다.

yfinance를 사용하여 일본의 Prime 시장, Standard 시장, Growth 시장의 최근 1년간 주가 데이터를 가져와 주세요.

그 후, 추이를 비교하기 쉽게 하기 위해 「시작일의 가격을 100」으로 하여 정규화한 뒤, 3개의 시장을 비교하는 꺾은선 그래프를 그려 주세요. 그래프의 일본어가 글자가 깨지지 않도록 대책도 부탁드립니다.

AI가 생성한 코드

import yfinance as yf
import pandas as pd
import matplotlib.pyplot as plt
...

그래프가 나오면 해상도를 높이기 위해 2차 분석을 실시하고 싶어지죠. 여기서 직접 프롬프트를 흘려보내도 되지만, 모처럼 AI가 제안해 주고 있으니 그것을 그대로 사용해 보겠습니다.

→ Gemini 1.5 Flash이므로, 2024년의 이란 전쟁 이야기를 가져와 버려서 몇 번의 채팅 대화를 실시.

→ 재미있는 결과가 나왔네요!!

Prime 시장은 매우 큰 영향이 나타난 반면, Standard 시장은 Prime에 비해 하락 폭이 작고, Growth 시장은 영향이 거의 받지 않은 것처럼 보인다.

Prime 시장에서 가장 큰 영향을 받은 것은 무엇인가? 어떤 업계인가? 등을 다음에 조사하고 싶어지네요. 여기서부터는 3차 분석, 4차 분석으로 이어갈 수 있습니다. (이번 내용과는 벗어나므로 생략합니다.)

과거에는 검증 환경을 갖추기 위해 VSCode를 열고, 패키지 충돌로 고민하며, 플롯 (Plot)의 미세 조정에 시간을 허비하곤 했습니다.

하지만 현재의 Google Colab은 **「사고의 속도를 늦추지 않는 샌드박스 (Sandbox)」**로서 완벽한 진화를 이루었습니다.

「코드를 작성하는 것」은 완전히 AI에게 맡기고, 인간은 **「데이터로부터 어떤 질문을 던지고, 어떤 시사점을 얻을 것인가」**라는 본질적인 인사이트 (Insight) 추출에 시간을 사용한다. 그런 AI 시대의 모던한 투자 분석 스타일을 여러분도 꼭 체험해 보시기 바랍니다.