Python提取通达信,自定义板块中成分股代码

get_initials("自选股")   # 返回 "zxg"

pip install pypinyin

# -*- coding: utf-8 -*-# @Author : 小红牛# 微信公众号：gxzfp888import osfrom pypinyin import pinyin, Styledef get_initials(text):    """将文本中的汉字转换为拼音首字母，非汉字直接保留"""    result = []    for char in text:        py = pinyin(char, style=Style.FIRST_LETTER)        if py and py[0]:            result.append(py[0][0].lower())        else:            result.append(char)    return ''.join(result)def read_tdx_block_codes(file_path):    """    读取通达信板块文件（.blk），提取每行中的股票代码（后6位数字）    参数:        file_path (str): 板块文件的完整路径    返回:        list: 股票代码字符串列表，例如 ['600519', '000001', ...]    """    codes = []    try:        with open(file_path, 'r', encoding='gbk') as f:            for line in f:                line = line.strip()                if not line:                    continue                # 取最后6位作为股票代码（兼容长度为6的行和含有前缀的行）                code = line[-6:] if len(line) >= 6 else line                codes.append(code)        return codes    except FileNotFoundError:        print(f"错误：文件未找到 - {file_path}")        return []    except Exception as e:        print(f"读取文件时出错：{e}")        return []# 使用示例if __name__ == "__main__":   # 改成你的通达信路径，和要查询的自定义板块，如自选股。    block = get_initials('自选股')        path = f'E:/new_tdx64/T0002/blocknew/{block}.blk'    print(block,path)    stock_codes = read_tdx_block_codes(path)    print("板块成分股如下：", stock_codes)