最近临时有个想法,输入股票代码, 输出股票基本信息,画出K线图,列出最新的新闻,并让AI大模型给出投资建议。
说干就干,刚折腾了一个小玩意, 先看图
这里简单提供下代码片段
1、获取股票基本信息
def get_stock_info(stock_code):
"""获取股票详细信息"""
try:
stock_info = ak.stock_individual_info_em(symbol=stock_code)
return stock_info
except Exception as e:
st.error(f"获取股票信息时出错: {str(e)}")
return None
2、画K线图
def get_stock_data(stock_code):
"""获取股票数据并绘制K线图"""
try:
# 获取3个月K线图
end_date = datetime.now().strftime('%Y%m%d')
start_date = (datetime.now() - timedelta(days=90)).strftime('%Y%m%d')
# 使用akshare获取股票数据
df = ak.stock_z