From 5a0519beee92fb739c6ee54c80c91521f5e2b7a5 Mon Sep 17 00:00:00 2001
From: guanjihuan <guanjihuan@163.com>
Date: Thu, 28 Dec 2023 20:21:54 +0800
Subject: [PATCH] Create stock_symbols_classification.py

---
 .../stock_symbols_classification.py           | 85 +++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 2023.12.28_stock_symbols_classification/stock_symbols_classification.py

diff --git a/2023.12.28_stock_symbols_classification/stock_symbols_classification.py b/2023.12.28_stock_symbols_classification/stock_symbols_classification.py
new file mode 100644
index 0000000..5feb23f
--- /dev/null
+++ b/2023.12.28_stock_symbols_classification/stock_symbols_classification.py
@@ -0,0 +1,85 @@
+"""
+This code is supported by the website: https://www.guanjihuan.com
+The newest version of this code is on the web page: https://www.guanjihuan.com/archives/38180
+"""
+
+import akshare as ak
+import re
+
+stocks = ak.stock_zh_a_spot_em()
+stock_data = stocks.values
+stock_symbols = stock_data[:, 1]
+num_stocks = len(stock_symbols)
+print('所有股票数量：', num_stocks)
+# print(stock_symbols)
+
+# 上交所主板
+stock_symbols_60 = []
+for stock_symbol in stock_symbols:
+    find_600 = re.findall(r'^600', stock_symbol)
+    find_601 = re.findall(r'^601', stock_symbol)
+    find_603 = re.findall(r'^603', stock_symbol)
+    find_605 = re.findall(r'^605', stock_symbol)
+    if find_600 != [] or find_601 != [] or find_603 != [] or find_605 != []:
+        stock_symbols_60.append(stock_symbol)
+num_stocks_60 = len(stock_symbols_60)
+print('上交所主板股票数量：', num_stocks_60)
+# print(stock_symbols_60)
+
+# 深交所主板
+stock_symbols_00 = []
+for stock_symbol in stock_symbols:
+    find_000 = re.findall(r'^000', stock_symbol)
+    find_001 = re.findall(r'^001', stock_symbol)
+    find_002 = re.findall(r'^002', stock_symbol)
+    find_003 = re.findall(r'^003', stock_symbol)
+    if find_000 != [] or find_001 != [] or find_002 != [] or find_003 != []:
+        stock_symbols_00.append(stock_symbol)
+num_stocks_00 = len(stock_symbols_00)
+print('深交所主板股票数量：', num_stocks_00)
+# print(stock_symbols_00)
+
+# 创业板
+stock_symbols_30 = []
+for stock_symbol in stock_symbols:
+    find_300 = re.findall(r'^300', stock_symbol)
+    find_301 = re.findall(r'^301', stock_symbol)
+    if find_300 != [] or find_301 != []:
+        stock_symbols_30.append(stock_symbol)
+num_stocks_30 = len(stock_symbols_30)
+print('创业板股票数量：', num_stocks_30)
+# print(stock_symbols_30)
+
+# 科创板
+stock_symbols_68 = []
+for stock_symbol in stock_symbols:
+    find_688 = re.findall(r'^688', stock_symbol)
+    find_689 = re.findall(r'^689', stock_symbol)
+    if find_688 != [] or find_689 != []:
+        stock_symbols_68.append(stock_symbol)
+num_stocks_68= len(stock_symbols_68)
+print('科创板股票数量：', num_stocks_68)
+# print(stock_symbols_68)
+
+# 新三板
+stock_symbols_8_4 = []
+for stock_symbol in stock_symbols:
+    find_82 = re.findall(r'^82', stock_symbol)
+    find_83 = re.findall(r'^83', stock_symbol)
+    find_87 = re.findall(r'^87', stock_symbol)
+    find_88 = re.findall(r'^88', stock_symbol)
+    find_430 = re.findall(r'^430', stock_symbol)
+    find_420 = re.findall(r'^420', stock_symbol)
+    find_400 = re.findall(r'^400', stock_symbol)
+    if find_82 != [] or find_83 != [] or find_87 != [] or find_88 != [] or find_430 != [] or find_420 != [] or find_400 != []:
+        stock_symbols_8_4.append(stock_symbol)
+num_stocks_8_4= len(stock_symbols_8_4)
+print('新三板股票数量：', num_stocks_8_4)
+# print(stock_symbols_8_4)
+
+print('所有股票数量：', num_stocks_60+num_stocks_00+num_stocks_30+num_stocks_68+num_stocks_8_4)
+
+# 检查遗漏的股票代码
+for stock_symbol in stock_symbols:
+    if stock_symbol not in stock_symbols_60 and stock_symbol not in stock_symbols_00 and stock_symbol not in stock_symbols_30 and stock_symbol not in stock_symbols_68 and stock_symbol not in stock_symbols_8_4:
+        print(stock_symbol)
\ No newline at end of file