[파이썬] 티커 입력받아 주식 과거 데이터 주가 다운로드

 

파이썬으로 주식 과거 데이터 다운로드를 해봅시다.

import yfinance as yf

def save_stock_data_to_excel(ticker: str, file_name: str):
    try:
        # 주식 데이터를 가져옵니다.
        stock = yf.Ticker(ticker)

        # 과거 최대치 데이터를 다운로드합니다.
        data = stock.history(period='max')

        # 시간대 정보를 제거합니다.
        data.index = data.index.tz_localize(None)

        # 데이터프레임을 엑셀 파일로 저장합니다.
        data.to_excel(file_name, sheet_name=ticker)
        print(f"Data for {ticker} saved to {file_name}")
    except Exception as e:
        print(f"An error occurred: {e}")

# 사용자로부터 티커 입력 받기
ticker_symbol = input("Enter the stock ticker symbol (e.g., AAPL for US stocks or 005930 for Korean stocks): ")

# 한국 주식이면 KOSPI/KOSDAQ 형식으로 변경
if ticker_symbol.isdigit() and len(ticker_symbol) == 6:
    # KOSPI로 가정하고 .KS 추가 (필요에 따라 .KQ로 변경 가능)
    ticker_symbol += ".KS"

# 파일 이름 생성
file_name = f"{ticker_symbol}_historical_data.xlsx"

# 주가 데이터를 엑셀로 저장
save_stock_data_to_excel(ticker_symbol, file_name)