銘柄一覧を取得する (/equities/master)

1
import os
2
import requests
3
import pandas as pd
4

5
BASE_URL = "https://api.jquants.com/v2"
6

7

8
def fetch_listed_info(headers: dict) -> pd.DataFrame:
9
    """上場銘柄一覧を DataFrame として取得する。"""
10
    url = f"{BASE_URL}/equities/master"
11
    response = requests.get(url, headers=headers, timeout=30)
12
    response.raise_for_status()
13
    payload = response.json()
14
    return pd.DataFrame(payload["data"])
15

16

17
# API キーは環境変数から読む(ハードコード禁止)
18
api_key = os.environ["JQUANTS_API_KEY"]
19
headers = {"x-api-key": api_key}
20

21
listed = fetch_listed_info(headers)
22
print(f"取得件数: {len(listed)}")
23
print(listed.head())

1
prime = listed[listed["MktNm"] == "プライム"]
2
print(f"プライム銘柄数: {len(prime)}")

1
target_markets = ["プライム", "グロース"]
2
filtered = listed[listed["MktNm"].isin(target_markets)]
3
print(f"対象市場の銘柄数: {len(filtered)}")

1
info_comms = listed[listed["S17Nm"] == "情報通信・サービスその他"]
2
print(f"件数: {len(info_comms)}")
3
print(info_comms[["Code", "CoName", "MktNm"]].head())

1
# "銀行" を含む会社
2
banks = listed[listed["CoName"].str.contains("銀行", na=False)]
3
print(banks[["Code", "CoName"]].head())

1
def fetch_listed_info_at(headers: dict, date: str) -> pd.DataFrame:
2
    """指定した基準日の上場銘柄一覧を取得する。"""
3
    url = f"{BASE_URL}/equities/master"
4
    response = requests.get(url, headers=headers, params={"date": date}, timeout=30)
5
    response.raise_for_status()
6
    return pd.DataFrame(response.json()["data"])
7

8

9
snapshot = fetch_listed_info_at(headers, "2024-01-04")
10
print(f"2024-01-04 時点の銘柄数: {len(snapshot)}")

1
from pathlib import Path
2

3
cache_dir = Path("data/jquants")
4
cache_dir.mkdir(parents=True, exist_ok=True)
5
listed.to_parquet(cache_dir / "listed_info_latest.parquet", index=False)

J-Quants API の /equities/master から取得した DataFrame を受け取り、
次の条件で銘柄を抽出する関数 filter_universe(listed, markets, sectors) を書いてください。

引数:
- listed: 上場銘柄の DataFrame
- markets: 市場区分名のリスト(例: ["プライム"])
- sectors: 17 業種名のリスト(空リストなら絞らない)

戻り値: フィルタ後の DataFrame(列順は元のまま)

要件:
- 引数が空のリストならその条件で絞らない
- pandas 2.2 系の API を使う
- docstring を日本語で書く