Python实现程序化中文选股，附完整代码【建议收藏】

选股介绍

相对于小白来说想通过Python实现自动化选股还是比较麻烦的，主要是处理逻辑十分繁琐，找了好久，总算找到一个开源的Python模块可以实现快速程序化选股，操作十分简单，只需要将你的选股逻辑用中文表达出来就行，核心是依赖同花顺 问财的API。

模块安装

pip install pywencai

完成代码

import os
import pywencai

def xg_wencai(query, perpage=200, ret='df'):
    """
    使用爱问财API进行选股查询，并返回结果。

    :param query: 查询条件字符串
    :param perpage: 每页返回的结果数量，默认为200
    :param ret: 返回结果的格式，'df'表示返回DataFrame，'list'表示返回股票代码列表，默认为'df'
    :return: 查询结果的DataFrame或股票代码列表
    """
    try:
        df = pywencai.get(query=query, sort_key='股票代码', sort_order='asc', perpage=perpage)
        if df is None or df.empty:
            print("No data found for the given query.")
            return None
        
        df['code'] = df['股票代码'].str[:6]
        
        if ret == 'list':
            return df['code'].tolist()
        elif ret == 'df':
            return df.round(3)
        else:
            raise ValueError("Invalid value for 'ret'. Must be 'df' or 'list'.")
    except Exception as e:
        print(f"An error occurred: {e}")
        return None

if __name__ == '__main__':
    # 设置你的选股条件
    query = '量比大于5'  
    result = xg_wencai(query, perpage=200, ret='df')
    if result is not None:
        result.to_excel('量比大于5选股结果.xlsx')

问题交流

如果有什么问题也可以主页联系我交流，目前我使用是正常的