MultiFactor 0.3.1__tar.gz → 0.3.3__tar.gz

{multifactor-0.3.1 → multifactor-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MultiFactor
-Version: 0.3.1
+Version: 0.3.3
 Summary: 멀티팩터 기반 주식 데이터 수집 패키지
 Author-email: HANHO <hanhoman@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -14,6 +14,12 @@ Requires-Dist: pandas
 Requires-Dist: numpy
 Requires-Dist: matplotlib
 Requires-Dist: yfinance
+Requires-Dist: requests
+Requires-Dist: seaborn
+Requires-Dist: plotly
+Requires-Dist: ipywidgets
+Requires-Dist: ipython
+Requires-Dist: anywidget
 
 # MultiFactor 📈
 
@@ -105,7 +111,7 @@ mf_us = MultiFactorUS(N=50)
 시각화 분석 객체는 다음 코드로 생성합니다. 
 
 ```python
-visual = Visual()  
+vis = Visual()  
 ```
 
 <br>
@@ -1231,6 +1237,7 @@ mf_us.get_Ngroup(df, Ngroup=10)
 | 누적 수익률   | 미국주식 누적수익률 비교 차트            | vis.ret_us         |
 | 실적재무 종합  | 국내주식 실적 및 밸류 추이             | vis.finance_kr     |
 | 업종종목 트리맵 | 미국주식 주요종목 주가등락율 트리맵         | vis.treemap_us     |
+| 업종종목 트리맵 | 국내주식 주요종목 주가등락율 트리맵         | vis.treemap_kr     |
 | 수익률 히트맵  | 지수 연월 수익률 히트맵               | vis.heatmap_idx    |
 | 수익률 히트맵  | 국내주식 연월 수익률 히트맵             | vis.heatmap_kr     |
 | 수익률 히트맵  | 미국주식 연월 수익률 히트맵             | vis.heatmap_us     |
@@ -1336,6 +1343,16 @@ vis.treemap_us()
 
 <br>
 
+⑩ (업종종목 트리맵) 국내주식 주요종목 주가등락율 트리맵
+
+구글 Colab에서 다음 코드로 국내주식 주가등락률 트리맵을 출력합니다.
+
+```python
+vis.treemap_kr()
+```
+
+<br>
+
 ⑩ (수익률 히트맵) 지수 연월 수익률 히트맵
 
 ```python

{multifactor-0.3.1 → multifactor-0.3.3}/README.md

@@ -88,7 +88,7 @@ mf_us = MultiFactorUS(N=50)
 시각화 분석 객체는 다음 코드로 생성합니다. 
 
 ```python
-visual = Visual()  
+vis = Visual()  
 ```
 
 <br>
@@ -1214,6 +1214,7 @@ mf_us.get_Ngroup(df, Ngroup=10)
 | 누적 수익률   | 미국주식 누적수익률 비교 차트            | vis.ret_us         |
 | 실적재무 종합  | 국내주식 실적 및 밸류 추이             | vis.finance_kr     |
 | 업종종목 트리맵 | 미국주식 주요종목 주가등락율 트리맵         | vis.treemap_us     |
+| 업종종목 트리맵 | 국내주식 주요종목 주가등락율 트리맵         | vis.treemap_kr     |
 | 수익률 히트맵  | 지수 연월 수익률 히트맵               | vis.heatmap_idx    |
 | 수익률 히트맵  | 국내주식 연월 수익률 히트맵             | vis.heatmap_kr     |
 | 수익률 히트맵  | 미국주식 연월 수익률 히트맵             | vis.heatmap_us     |
@@ -1319,6 +1320,16 @@ vis.treemap_us()
 
 <br>
 
+⑩ (업종종목 트리맵) 국내주식 주요종목 주가등락율 트리맵
+
+구글 Colab에서 다음 코드로 국내주식 주가등락률 트리맵을 출력합니다.
+
+```python
+vis.treemap_kr()
+```
+
+<br>
+
 ⑩ (수익률 히트맵) 지수 연월 수익률 히트맵
 
 ```python

{multifactor-0.3.1 → multifactor-0.3.3}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 # 1. PyPI에 등록될 배포 이름 (pip install MultiFactor)
 name = "MultiFactor"
-version = "0.3.1"
+version = "0.3.3"
 authors = [
   { name="HANHO", email="hanhoman@gmail.com" },
 ]
@@ -27,6 +27,12 @@ dependencies = [
 	"numpy",	
 	"matplotlib",
 	"yfinance",
+  "requests",
+  "seaborn",
+  "plotly",
+  "ipywidgets",
+  "ipython",
+  "anywidget",
 ]
 
 # 3. src 폴더 안의 패키지를 자동으로 찾도록 설정

multifactor-0.3.3/src/MultiFactor/Visual.py

@@ -0,0 +1,94 @@
+import importlib
+import os
+
+from . import vmulti_idx
+from . import vmulti_kr
+from . import vmulti_us
+from . import vcompare_idx
+from . import vret_idx
+from . import vret_kr
+from . import vret_us
+from . import vfinance_kr
+from . import vheatmap_idx
+from . import vtreemap_us
+from . import vtreemap_kr
+from . import vheatmap_kr
+from . import vheatmap_us
+from . import vtargetprice_us
+
+
+def _set_korean_font():
+    try:
+        fm = importlib.import_module("matplotlib.font_manager")
+        plt = importlib.import_module("matplotlib.pyplot")
+    except ImportError:
+        return None
+
+    font_paths = [
+        "/usr/share/fonts/truetype/nanum/NanumGothic.ttf",
+        "/usr/share/fonts/truetype/nanum/NanumBarunGothic.ttf",
+        "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
+        "/usr/share/fonts/opentype/noto/NotoSansCJKkr-Regular.otf",
+        "C:/Windows/Fonts/malgun.ttf",
+        "C:/Windows/Fonts/malgunbd.ttf",
+        "/System/Library/Fonts/AppleSDGothicNeo.ttc",
+        "/Library/Fonts/AppleGothic.ttf",
+    ]
+
+    for font_path in font_paths:
+        if os.path.exists(font_path):
+            fm.fontManager.addfont(font_path)
+            font_name = fm.FontProperties(fname=font_path).get_name()
+            plt.rcParams["font.family"] = font_name
+            plt.rcParams["axes.unicode_minus"] = False
+            return font_name
+
+    plt.rcParams["axes.unicode_minus"] = False
+    return None
+
+
+class Visual:
+    def __init__(self):
+        self.korean_font = _set_korean_font()
+
+    def multi_idx(self):
+        vmulti_idx.multi_idx()
+
+    def multi_kr(self):
+        vmulti_kr.multi_kr()
+        
+    def multi_us(self):
+        vmulti_us.multi_us()
+
+    def compare_idx(self):
+        vcompare_idx.compare_idx()
+
+    def ret_idx(self):
+        vret_idx.ret_idx()
+
+    def ret_kr(self):
+        vret_kr.ret_kr()
+        
+    def ret_us(self):
+        vret_us.ret_us()
+
+    def finance_kr(self):
+        vfinance_kr.finance_kr()
+
+    def heatmap_idx(self):
+        vheatmap_idx.heatmap_idx()
+
+    def treemap_us(self):
+        vtreemap_us.treemap_us()
+
+    def treemap_kr(self):
+        vtreemap_kr.treemap_kr()
+
+    def heatmap_kr(self):
+        vheatmap_kr.heatmap_kr()
+
+    def heatmap_us(self):
+        vheatmap_us.heatmap_us()
+
+    def targetprice_us(self):
+        vtargetprice_us.targetprice_us()

{multifactor-0.3.1 → multifactor-0.3.3}/src/MultiFactor/__init__.py

@@ -15,5 +15,7 @@ from .core import MultiFactorKR
 from .us_core import MultiFactorUS
 from .Visual import Visual
 
+__version__ = "0.3.3"
+
 # 패키지를 import할 때 노출할 항목을 명시합니다.
-__all__ = ['MultiFactorKR', 'MultiFactorUS', 'Visual']
+__all__ = ['MultiFactorKR', 'MultiFactorUS', 'Visual', '__version__']

multifactor-0.3.3/src/MultiFactor/vtreemap_kr.py

@@ -0,0 +1,130 @@
+from io import StringIO
+
+import numpy as np
+import pandas as pd
+import FinanceDataReader as fdr
+import plotly.express as px
+import plotly.graph_objects as go
+import ipywidgets as widgets
+from IPython.display import clear_output, display
+
+
+def _enable_colab_widgets():
+    try:
+        from google.colab import output
+    except ImportError:
+        return False
+
+    output.enable_custom_widget_manager()
+    return True
+
+
+def _generate_sector_data(prompt):
+    try:
+        from google.colab import ai
+    except ImportError as exc:
+        raise RuntimeError("google.colab.ai를 사용할 수 있는 구글 Colab 환경에서 실행해주세요.") from exc
+
+    response = ai.generate_text(prompt)
+    sector = pd.read_csv(StringIO(response), names=["ticker_name", "sector"], on_bad_lines="skip")
+    sector["ticker_name"] = sector["ticker_name"].astype(str).str.strip()
+    sector["sector"] = sector["sector"].astype(str).str.strip()
+    sector = sector[(sector["ticker_name"] != "") & (sector["sector"] != "")]
+    return sector.drop_duplicates(subset=["ticker_name"], keep="first")
+
+
+def treemap_kr():
+    _enable_colab_widgets()
+
+    print("코스피 주식 데이터를 가져오는 중입니다...")
+    required_cols = ["Code", "Name", "Marcap", "ChagesRatio", "Close"]
+    df_base = fdr.StockListing("KOSPI")
+
+    missing_cols = [col for col in required_cols if col not in df_base.columns]
+    if missing_cols:
+        print(f"필수 컬럼을 찾을 수 없습니다: {', '.join(missing_cols)}")
+        return
+
+    df_base = df_base[required_cols].copy()
+    df_base.columns = ["ticker", "ticker_name", "mktcap", "ret", "close"]
+    df_base["ticker_name"] = df_base["ticker_name"].astype(str).str.strip()
+    df_base = df_base.dropna(subset=["ticker_name", "mktcap", "ret", "close"])
+    df_base = df_base[df_base["mktcap"] > 0]
+
+    if df_base.empty:
+        print("트리맵을 생성할 코스피 데이터를 찾을 수 없습니다.")
+        return
+
+    print("데이터 로드 완료. 아래 드롭다운에서 분석할 종목 수를 선택하세요.")
+
+    @widgets.interact(top_n=widgets.Dropdown(options=[50, 100, 200], value=100, description="종목 수:"))
+    def generate_stock_treemap(top_n):
+        df = df_base.sort_values(by="mktcap", ascending=False).head(top_n).copy()
+        stock_names = "\n".join(df["ticker_name"])
+
+        prompt = f"""
+#지시문
+- 다음 종목명의 종목섹터를 알고 싶습니다.
+- 종목섹터는 한글로 작성해주세요.
+- 종목별로 섹터가 여러개로 분류되면, 대표섹터 하나만 생성합니다. (예: 반도체/IT이면, 반도체만 생성)
+- 출력형식을 정확히 지켜서 작성해주세요. 형식이 틀리면 안 됩니다.
+
+#출력형식
+<종목명>,<섹터명>
+<종목명>,<섹터명>
+
+#종목명
+{stock_names}
+"""
+
+        max_retries = 3
+        sector = pd.DataFrame(columns=["ticker_name", "sector"])
+
+        for retry_count in range(max_retries):
+            try:
+                print(f"\r[Gemini API] 업종 정보 추론 중... (시도 {retry_count + 1}/{max_retries})", end="")
+                sector_temp = _generate_sector_data(prompt)
+                test_merge = df.merge(sector_temp, on="ticker_name", how="left")
+                nan_count = test_merge["sector"].isna().sum() + (test_merge["sector"] == "nan").sum()
+
+                if nan_count < len(df) * 0.1:
+                    sector = sector_temp
+                    print(" -> 성공!")
+                    break
+            except Exception as exc:
+                if retry_count == max_retries - 1:
+                    print(f"\n[오류] 업종 정보 추론에 실패했습니다. 기타 업종으로 진행합니다. ({exc})")
+
+        df_final = df.merge(sector, on="ticker_name", how="left")
+        df_final["sector"] = df_final["sector"].fillna("기타 업종").replace("nan", "기타 업종")
+        df_final["mktcap_log"] = np.log10(df_final["mktcap"])
+
+        custom_colorscale = [
+            [0.0, "rgb(44, 95, 233)"],
+            [0.4, "rgb(151, 185, 255)"],
+            [0.5, "rgb(255, 255, 255)"],
+            [0.6, "rgb(255, 165, 165)"],
+            [1.0, "rgb(235, 41, 41)"],
+        ]
+
+        fig = px.treemap(
+            df_final,
+            path=["sector", "ticker_name"],
+            values="mktcap_log",
+            color="ret",
+            color_continuous_scale=custom_colorscale,
+            color_continuous_midpoint=0,
+            title=f"코스피 시가총액 상위 {top_n} 종목 업종/종목별 등락률 트리맵",
+            hover_data={"ret": ":.2f", "mktcap": ":,", "close": ":,", "ticker": True, "mktcap_log": False},
+        )
+
+        fig.update_layout(
+            width=1000,
+            height=700,
+            title_font_size=16,
+            title_font_family="Malgun Gothic",
+            margin=dict(t=50, l=10, r=10, b=10),
+        )
+
+        clear_output(wait=True)
+        display(go.FigureWidget(fig))

{multifactor-0.3.1 → multifactor-0.3.3}/src/MultiFactor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MultiFactor
-Version: 0.3.1
+Version: 0.3.3
 Summary: 멀티팩터 기반 주식 데이터 수집 패키지
 Author-email: HANHO <hanhoman@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -14,6 +14,12 @@ Requires-Dist: pandas
 Requires-Dist: numpy
 Requires-Dist: matplotlib
 Requires-Dist: yfinance
+Requires-Dist: requests
+Requires-Dist: seaborn
+Requires-Dist: plotly
+Requires-Dist: ipywidgets
+Requires-Dist: ipython
+Requires-Dist: anywidget
 
 # MultiFactor 📈
 
@@ -105,7 +111,7 @@ mf_us = MultiFactorUS(N=50)
 시각화 분석 객체는 다음 코드로 생성합니다. 
 
 ```python
-visual = Visual()  
+vis = Visual()  
 ```
 
 <br>
@@ -1231,6 +1237,7 @@ mf_us.get_Ngroup(df, Ngroup=10)
 | 누적 수익률   | 미국주식 누적수익률 비교 차트            | vis.ret_us         |
 | 실적재무 종합  | 국내주식 실적 및 밸류 추이             | vis.finance_kr     |
 | 업종종목 트리맵 | 미국주식 주요종목 주가등락율 트리맵         | vis.treemap_us     |
+| 업종종목 트리맵 | 국내주식 주요종목 주가등락율 트리맵         | vis.treemap_kr     |
 | 수익률 히트맵  | 지수 연월 수익률 히트맵               | vis.heatmap_idx    |
 | 수익률 히트맵  | 국내주식 연월 수익률 히트맵             | vis.heatmap_kr     |
 | 수익률 히트맵  | 미국주식 연월 수익률 히트맵             | vis.heatmap_us     |
@@ -1336,6 +1343,16 @@ vis.treemap_us()
 
 <br>
 
+⑩ (업종종목 트리맵) 국내주식 주요종목 주가등락율 트리맵
+
+구글 Colab에서 다음 코드로 국내주식 주가등락률 트리맵을 출력합니다.
+
+```python
+vis.treemap_kr()
+```
+
+<br>
+
 ⑩ (수익률 히트맵) 지수 연월 수익률 히트맵
 
 ```python

{multifactor-0.3.1 → multifactor-0.3.3}/src/MultiFactor.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ src/MultiFactor/vret_idx.py
 src/MultiFactor/vret_kr.py
 src/MultiFactor/vret_us.py
 src/MultiFactor/vtargetprice_us.py
+src/MultiFactor/vtreemap_kr.py
 src/MultiFactor/vtreemap_us.py
 src/MultiFactor.egg-info/PKG-INFO
 src/MultiFactor.egg-info/SOURCES.txt

multifactor-0.3.3/src/MultiFactor.egg-info/requires.txt

@@ -0,0 +1,11 @@
+finance-datareader
+pandas
+numpy
+matplotlib
+yfinance
+requests
+seaborn
+plotly
+ipywidgets
+ipython
+anywidget

multifactor-0.3.1/src/MultiFactor/Visual.py

@@ -1,54 +0,0 @@
-from . import vmulti_idx
-from . import vmulti_kr
-from . import vmulti_us
-from . import vcompare_idx
-from . import vret_idx
-from . import vret_kr
-from . import vret_us
-from . import vfinance_kr
-from . import vheatmap_idx
-from . import vtreemap_us
-from . import vheatmap_kr
-from . import vheatmap_us
-from . import vtargetprice_us
-
-
-class Visual:
-    def multi_idx(self):
-        vmulti_idx.multi_idx()
-
-    def multi_kr(self):
-        vmulti_kr.multi_kr()
-        
-    def multi_us(self):
-        vmulti_us.multi_us()
-
-    def compare_idx(self):
-        vcompare_idx.compare_idx()
-
-    def ret_idx(self):
-        vret_idx.ret_idx()
-
-    def ret_kr(self):
-        vret_kr.ret_kr()
-        
-    def ret_us(self):
-        vret_us.ret_us()
-
-    def finance_kr(self):
-        vfinance_kr.finance_kr()
-
-    def heatmap_idx(self):
-        vheatmap_idx.heatmap_idx()
-
-    def treemap_us(self):
-        vtreemap_us.treemap_us()
-
-    def heatmap_kr(self):
-        vheatmap_kr.heatmap_kr()
-
-    def heatmap_us(self):
-        vheatmap_us.heatmap_us()
-
-    def targetprice_us(self):
-        vtargetprice_us.targetprice_us()

multifactor-0.3.1/src/MultiFactor.egg-info/requires.txt

@@ -1,5 +0,0 @@
-finance-datareader
-pandas
-numpy
-matplotlib
-yfinance