首页量化学习正文

Python自动化炒股：使用Streamlit和Heroku部署股票数据分析仪表盘的实战案例

量化学习 2024-06-20 3513

Python 自动化炒股：使用Streamlit和Heroku部署股票数据分析仪表盘的实战案例

引言

在当今的金融市场中，自动化交易和数据分析已经成为投资者和交易者的重要工具。Python以其强大的库和灵活性，成为了实现这些功能的首选语言。本文将带你了解如何使用Python、Streamlit和Heroku来创建并部署一个股票数据分析仪表盘，让你能够实时监控股票市场动态，并做出更明智的交易决策。

准备工作

在开始之前，确保你已经安装了Python环境，并准备好了以下工具和库：

Python 3.x
Streamlit
Pandas
NumPy
Matplotlib
YFinance（用于获取股票数据）
Heroku CLI（用于部署应用）

你可以通过以下命令安装所需的库：

pip install streamlit pandas numpy matplotlib yfinance

步骤1：获取股票数据

首先，我们需要获取股票数据。我们将使用yfinance库来获取数据。以下是获取特定股票数据的示例代码：

import yfinance as yf

def get_stock_data(ticker, period='1mo'):
    stock = yf.Ticker(ticker)
    hist = stock.history(period=period)
    return hist

# 示例：获取苹果公司股票数据
apple_stock_data = get_stock_data('AAPL')

步骤2：数据分析

获取数据后，我们需要进行一些基本的数据分析。这里我们将计算股票的移动平均线（MA）：

import pandas as pd

def calculate_ma(data, window):
    return data['Close'].rolling(window=window).mean()

# 计算10日和50日移动平均线
apple_stock_data['MA10'] = calculate_ma(apple_stock_data, 10)
apple_stock_data['MA50'] = calculate_ma(apple_stock_data, 50)

步骤3：创建Streamlit应用

现在，我们将使用Streamlit来创建一个简单的股票数据分析仪表盘。以下是一个基本的Streamlit应用示例：

import streamlit as st
import matplotlib.pyplot as plt

def plot_stock_data(data):
    fig, ax = plt.subplots()
    ax.plot(data['Close'], label='Close Price')
    ax.plot(data['MA10'], label='10-Day MA')
    ax.plot(data['MA50'], label='50-Day MA')
    ax.set_title('Stock Price and Moving Averages')
    ax.set_xlabel('Date')
    ax.set_ylabel('Price')
    ax.legend()
    plt.xticks(rotation=45)
    st.pyplot(fig)

def mAIn():
    st.title('Stock Data Analysis Dashboard')
    
    ticker = st.text_input('Enter a stock ticker:', 'AAPL')
    period = st.selectbox('Select a period:', ['1mo', '3mo', '6mo', '1y', '2y', '5y', '10y'])
    
    stock_data = get_stock_data(ticker, period)
    plot_stock_data(stock_data)

if __name__ == '__main__':
    main()

步骤4：本地测试

在部署之前，你应该在本地测试你的Streamlit应用。运行以下命令来启动应用：

streamlit run your_script.py

确保一切正常工作后，我们可以进行下一步。

步骤5：部署到Heroku

为了将你的Streamlit应用部署到Heroku，你需要一个Heroku账户和一个Git仓库。以下是部署步骤：

在Heroku创建一个新的应用。
将你的代码推送到Git仓库。
使用Heroku CLI连接你的Git仓库和Heroku应用。
配置Heroku环境变量。
部署你的应用。

以下是部署的示例命令：

# 登录Heroku CLI
heroku login

# 创建一个新的Heroku应用
heroku create your-app-name

# 初始化Git仓库（如果尚未初始化）
git init

# 添加远程仓库
git remote add heroku https://git.heroku.com/your-app-name.git

# 推送代码到Heroku
git push heroku master

# 打开应用
heroku open