江苏龙网

  1. 主页 > 生活百科 > >

智能决策新纪元：使用AutoGen分析金融数据( 三 )

AutoGen

3.初始化聊天和发送消息
# user_proxy使用initiate_chat方法向助手发起聊天，并发送一条消息 。user_proxy.initiate_chat(assistant,# 指定要发送到的助手 。message="""今天是几号? 请帮我比较万科A股票和招商银行股票的收益情况，用图表的形式对两者进行比较 。""",# 发送的消息内容 。)assistant 是一个配置了特定参数的助手代理，可能用于执行某种自动化任务或处理 。
user_proxy 是一个模拟用户的代理，配置了自动回复和终止对话的条件 。
user_proxy 通过 initiate_chat 方法与 assistant 开启对话，并发送了一个关于比较股票收益的任务描述 。
结果展示在执行代码之后可以产生结果，由于结果内容比较长，涉及到方案的提出，代码生成，代码验证，代码修改等过程 。这是一个复杂的自我修正过程，主要体现了用户代理与助手之间的互动中，我把结果的输出整理成如下内容，方便大家阅读：
提问-用户代理To助手:用户询问了当前的日期，并请求比较万科A股票与招商银行股票的收益情况，并要求以图表形式展示 。
得出方案-助手To用户代理:助手提出了一个解决方案，该方案分为三个主要步骤：
1. 使用Python的datetime库获取当前日期 。
2. 利用pandas_datareader库从Yahoo Finance获取股票的历史数据 。
3. 使用matplotlib库将数据绘制成图表形式 ， 比较两只股票的收益情况 。
然而，当尝试执行获取Yahoo财经数据的代码时遇到了问题 ， 错误提示表明类型错误，字符串索引必须是整数 。
再次执行代码-助手To用户代理:助手建议使用yfinance库来替代pandas_datareader，以解决从Yahoo财经获取数据的问题 。在成功安装yfinance库后，助手提供了新的代码片段来重新获取股票数据，并将数据保存到CSV文件中 。然后，又提供了另一段代码来创建图表，这次的代码执行成功 。
最终得到结果-助手To用户代理:助手确认万科A和招商银行的历史价格数据已成功获取 ， 并且图表已创建 。由于环境限制，助手指出无法直接显示图表，并建议用户在本地环境中运行代码以查看图表 。最终，通过发送“TERMINATE”，结束了对话 。
AutoGen任务代码在整个过程中，用户代理与助手之间的交互主要集中在解决问题和代码执行上 。助手在诊断并解决问题时表现出适应性和灵活性，最终提供了满足用户请求的结果 。
下面我们把用户助理生成的代码放在下面，大家可以参考 。这段Python代码用于比较两只股票（万科A和招商银行）的每日收益情况，并将结果以图表形式展示 。下面是对每一部分代码的详细解释：
# 导入必要的Python库 。import datetime# 用于处理日期和时间 。import pandas as pd# 用于数据分析和操作 。import matplotlib.pyplot as plt# 用于数据可视化 。import yfinance as yf# 用于从Yahoo Finance下载股票数据 。# 获取当前日期并打印 。today = datetime.date.today()print("今天是：", today)# 设置获取股票数据的起始和结束日期 。start_date = '2020-01-01'# 设置起始日期为2020年1月1日 。end_date = today# 设置结束日期为当前日期 。# 使用yfinance下载万科A和招商银行的股票数据 。vanke = yf.download('000002.SZ', start=start_date, end=end_date)# 下载万科A的数据 。cmb = yf.download('600036.SS', start=start_date, end=end_date)# 下载招商银行的数据 。# 计算每日收益率 。收益率是通过将每日的收盘价与前一日的收盘价进行比较计算得出的 。vanke['Daily Return'] = vanke['Close'].pct_change()# 计算万科A的每日收益率 。cmb['Daily Return'] = cmb['Close'].pct_change()# 计算招商银行的每日收益率 。# 使用matplotlib创建图表并设置图表大小 。plt.figure(figsize=(12,6))# 设置图表的标题 。plt.title('Daily Return Comparison')# 绘制两只股票的每日收益率曲线 。vanke['Daily Return'].plot(label='Vanke A')# 绘制万科A的收益率曲线 。cmb['Daily Return'].plot(label='CMB')# 绘制招商银行的收益率曲线 。# 添加图例 。plt.legend()# 显示图表 。plt.show()代码整体比较简单，其执行过程如下：
1. 导入所需的库 。
2. 获取并打印当前日期 。
3. 定义获取股票数据的时间范围 。
4. 从Yahoo Finance下载指定日期范围内的万科A和招商银行股票数据 。
5. 计算并添加每日收益率到下载的数据中 。
6. 绘制并显示两只股票的每日收益率对比图表 。
代码中使用的yfinance库是一个流行的金融数据接口 ， 可从Yahoo Finance下载历史市场数据 。Matplotlib是一个广泛使用的Python绘图库，可以创建多种静态动态和交互式图表 。.pct_change()函数用于计算数据帧中元素的百分比变化，常用于金融数据分析中计算收益率 。plt.show()函数调用会打开一个窗口展示生成的图表 。在Jupyter Notebook或其他交互式环境中，图表通常会直接显示 。


推荐阅读


上一篇：Oracle软件在主机平台的应用

下一篇：Linux中Awk命令的五个常见用法