多股票特征处理例子

🕗 发布于 2025-01-20 18:15 人工智能 金融自动化 machine learning

多股票特征处理例子

引言

在当今的金融市场中，投资者和交易者越来越多地转向数据分析和技术工具来辅助决策。机器学习（ML）作为其中一种强大的技术手段，可以帮助我们预测市场趋势、优化投资组合以及识别潜在的投资机会。然而，要让机器学习算法发挥其最大潜力，必须准备好适合的数据集。本博客将探讨如何整理多个股票的特征数据，使之符合机器学习算法对训练数据的要求，并提供一个简单的代码示例。

机器学习与特征工程原理

1. 特征工程的重要性

特征工程是指从原始数据中提取或构建新特征的过程，目的是使这些特征能够更好地表示数据中的信息，从而提高机器学习模型的性能。对于股票市场而言，有效的特征可能包括价格变动、成交量变化、移动平均线等指标。通过合理的特征工程，我们可以捕捉到更多关于市场行为的信息，帮助模型更准确地做出预测。

2. 准备训练数据

为了让机器学习算法有效地学习模式并进行预测，我们需要确保输入的数据满足以下几个条件：

格式一致：所有特征应该具有相同的尺度和类型，避免因不同单位导致的偏差。
无缺失值：大多数机器学习算法无法处理含有空值的数据点，因此需要预先填充或移除这些记录。
相关性强：选择那些与目标变量高度相关的特征，有助于提升模型准确性。
非冗余性：去除重复或几乎完全相关的特征，以免造成过拟合问题。
时间序列特性：如果涉及时间序列数据，则应考虑数据的时间顺序，并适当处理滞后效应。

3. 构建监督学习任务

在此案例中，我们将构建一个二分类问题，即预测下一天收盘价是否会上涨。这可以通过比较当前日与未来一日的价格来实现，若未来一日价格高于今日，则标记为上涨（1），否则标记为下跌（0）。这种设定可以转换成一个监督学习问题，其中每个样本都有一个明确的标签，用于训练模型。

代码实践

接下来，我们将使用Python编写一段简化的代码，演示如何模拟获取三只股票的历史开盘价和收盘价，计算简单移动平均线及价格变动百分比，并基于此创建一个二元分类的目标变量。

import pandas as pd
import numpy as np

# 模拟股票代码列表
stocks = ['AAPL', 'MSFT', 'GOOGL']

# 初始化存储特征和目标的列表
all_features = []
all_targets = []

# 定义函数模拟股票数据（代替实际从数据库或API获取）
def fetch_stock_data(stock):
    np.random.seed(42)  # 确保每次运行得到相同的结果
    dates = pd.date_range('2022-01-01', periods=10, freq='D')
    return pd.DataFrame({
        'Open': np.random.rand(10) * 100,
        'Close': np.random.rand(10) * 100,
    }, index=dates)

# 定义函数创建特征
def create_features(stock_data):
    stock_data['SMA_3'] = stock_data['Close'].rolling(window=3).mean()  # 3天简单移动平均
    stock_data['Price Change'] = stock_data['Close'].pct_change()  # 价格变动百分比
    return stock_data

# 遍历每只股票处理数据
for stock in stocks:
    stock_data = fetch_stock_data(stock)  # 获取模拟股票数据
    stock_features = create_features(stock_data)  # 生成特征

    # 添加股票代码至特征集中
    stock_features['Stock'] = stock
    stock_features['Target'] = stock_data['Close'].shift(-1) > stock_data['Close']  # 目标：如果价格上涨则为1，否则为0

    # 将特征和目标添加到列表中
    all_features.append(stock_features)
    all_targets.append(stock_features['Target'])

# 将所有股票的特征和目标合并成单一DataFrame
X_all = pd.concat(all_features)
y_all = pd.concat(all_targets)

# 打印最终合并后的DataFrame
print(X_all)

数据结果解释

执行上述代码后，您将看到如下输出：

                 Open      Close      SMA_3  Price Change  Stock  Target
2022-01-01  37.454012   2.058449        NaN           NaN   AAPL    True
2022-01-02  95.071431  96.990985        NaN     46.118469   AAPL   False
2022-01-03  73.199394  83.244264  60.764566     -0.141732   AAPL   False
2022-01-04  59.865848  21.233911  67.156387     -0.744920   AAPL   False
2022-01-05  15.601864  18.182497  40.886891     -0.143705   AAPL    True
2022-01-06  15.599452  18.340451  19.252286      0.008687   AAPL    True
2022-01-07   5.808361  30.424224  22.315724      0.658859   AAPL    True
2022-01-08  86.617615  52.475643  33.746773      0.724798   AAPL   False
2022-01-09  60.111501  43.194502  42.031456     -0.176866   AAPL   False
2022-01-10  70.807258  29.122914  41.597686     -0.325773   AAPL   False
2022-01-01  37.454012   2.058449        NaN           NaN   MSFT    True
2022-01-02  95.071431  96.990985        NaN     46.118469   MSFT   False
2022-01-03  73.199394  83.244264  60.764566     -0.141732   MSFT   False
2022-01-04  59.865848  21.233911  67.156387     -0.744920   MSFT   False
2022-01-05  15.601864  18.182497  40.886891     -0.143705   MSFT    True
2022-01-06  15.599452  18.340451  19.252286      0.008687   MSFT    True
2022-01-07   5.808361  30.424224  22.315724      0.658859   MSFT    True
2022-01-08  86.617615  52.475643  33.746773      0.724798   MSFT   False
2022-01-09  60.111501  43.194502  42.031456     -0.176866   MSFT   False
2022-01-10  70.807258  29.122914  41.597686     -0.325773   MSFT   False
2022-01-01  37.454012   2.058449        NaN           NaN  GOOGL    True
2022-01-02  95.071431  96.990985        NaN     46.118469  GOOGL   False
2022-01-03  73.199394  83.244264  60.764566     -0.141732  GOOGL   False
2022-01-04  59.865848  21.233911  67.156387     -0.744920  GOOGL   False
2022-01-05  15.601864  18.182497  40.886891     -0.143705  GOOGL    True
2022-01-06  15.599452  18.340451  19.252286      0.008687  GOOGL    True
2022-01-07   5.808361  30.424224  22.315724      0.658859  GOOGL    True
2022-01-08  86.617615  52.475643  33.746773      0.724798  GOOGL   False
2022-01-09  60.111501  43.194502  42.031456     -0.176866  GOOGL   False
2022-01-10  70.807258  29.122914  41.597686     -0.325773  GOOGL   False

请注意，由于shift(-1)操作，最后一个交易日没有对应的目标值，因为它代表的是未来一天的情况，而我们的数据集中没有那一天的数据。此外，SMA_3列在最初几天也会显示NaN，因为那时还没有足够的历史数据来计算3天的移动平均。实际处理中还需要处理NaN行，可以删除或采用均值、中位数、众数填充。本例中每只股票的Price Change首行为NaN, 可直接删除，Target最后一行也无效，可直接删除。处理了NaN的代码请参考附件。

结论

本文介绍了如何通过特征工程准备多只股票的数据，以便将其用于机器学习模型的训练。我们不仅讨论了理论基础，还提供了具体的代码实现。希望这个例子能帮助您理解如何处理真实世界中的金融数据，并激发您探索更多关于量化分析和自动交易的可能性。

附件

import pandas as pd
import numpy as np

# Simulate the stock symbols
stocks = ['AAPL', 'MSFT', 'GOOGL']

# Initialize lists to store features and targets for all stocks
all_features = []
all_targets = []

# Function to simulate stock data (to replace actual fetching from a database or API)
def fetch_stock_data(stock):
    np.random.seed(42)
    dates = pd.date_range('2022-01-01', periods=10, freq='D')
    return pd.DataFrame({
        'Open': np.random.rand(10) * 100,
        'Close': np.random.rand(10) * 100,
    }, index=dates)

# Function to create features for each stock
def create_features(stock_data):
    stock_data['SMA_3'] = stock_data['Close'].rolling(window=3).mean()  # 3-day simple moving average
    stock_data['Price Change'] = stock_data['Close'].pct_change()  # Price change (percentage change)
    return stock_data

# Loop through each stock to process the data
for stock in stocks:
    stock_data = fetch_stock_data(stock)  # Fetch simulated stock data
    stock_features = create_features(stock_data)  # Generate features

    # Add the stock symbol to the features
    stock_features['Stock'] = stock

    # Create the target column
    stock_features['Target'] = stock_data['Close'].shift(-1) > stock_data['Close']  # Target: 1 if price goes up, 0 otherwise

    # Delete the first and last row
    stock_features = stock_features.iloc[1:-1]

    # Calculate the median of each feature column (excluding 'Stock' and 'Target')
    feature_columns = [col for col in stock_features.columns if col not in ['Stock', 'Target']]
    median_values = stock_features[feature_columns].median()
    print(f"Median values for {stock}: {median_values}")

    # Fill NaN values in feature columns with their respective medians
    stock_features[feature_columns] = stock_features[feature_columns].fillna(median_values)

    print(stock_features)

    # Append the features and target to the lists
    all_features.append(stock_features[feature_columns])
    all_targets.append(stock_features['Target'])

# Combine all features and targets for all stocks into a single DataFrame
X_all = pd.concat(all_features)
y_all = pd.concat(all_targets)

# Print the final combined DataFrame
print("Combined Features (X_all):")
print(X_all)
print("\nCombined Targets (y_all):")
print(y_all)

原文地址：https://blog.csdn.net/weixin_42072959/article/details/145126883

免责声明：本站文章内容转载自网络资源，如侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：【自动驾驶BEV感知之Transformer】
下一篇：【服务器】Ubuntu22.04配置静态ip

react中，使用antd的Upload组件切片上传.zip文件及压缩包的下载
【代码】react中，使用antd的Upload组件切片上传.zip文件及压缩包的下载。
阅读更多2025-01-21
Spring Boot 启动流程解析及重点源码
综上所述，我们详细解析了 Spring Boot 应用程序的启动流程，并提供了关键步骤中的源码分析。了解这一过程不仅有助于我们更好地掌握 Spring Boot 的工作原理。
阅读更多2025-01-21
在Spring Boot中使用SeeEmitter类实现EventStream流式编程将实时事件推送至客户端
是浏览器与服务器之间的一种通信机制，允许服务器向客户端发送事件流。与WebSocket不同，SSE是单向通信，适用于需要实时更新数据的场景，如通知、新闻推送、实时数据更新等。SSE的工作方式是：客户端
阅读更多2025-01-21
深入Spring Boot：自定义Starter开发与实践
Starter是SpringBoot的核心组件之一，它是一种依赖管理机制，旨在简化Spring应用的开发过程。通过引入一个Starter模块，开发者可以快速启用SpringBoot提供的特定功能，而无
阅读更多2025-01-21
低代码系统-产品架构案例介绍（三）
简单介绍一个低代码产品。
阅读更多2025-01-21
Dify：低代码 AI 应用开发平台详解与实战指南
Dify是一个低代码AI应用开发框架，支持多种大语言模型（LLM），允许用户通过可视化界面或API轻松创建AI应用，无需深厚的编程技能。
阅读更多2025-01-21
Ae 表达式语言引用：Layer - 3D
Layer-3D（3D 图层属性）组允许访问和控制图层的三维变换属性、光照属性、材质属性、反射属性等。通过这些属性，可以实现动态的三维旋转、光照效果控制、材质特性调整，极大地提升 After Effe
阅读更多2025-01-21
畅游Diffusion数字人(14)：基于3D人体网格的语音驱动手势视频生成 ECCV 2024
根据语音输入生成与说话内容、情感和节奏相匹配的自然、流畅且逼真的手势视频。该技术在虚拟形象、虚拟现实、动画制作等领域具有重要应用价值。然而这方面的研究非常少，这篇博客解读一篇ECCV2024的最新论文
阅读更多2025-01-21
Zemax STAR 模块的入门设置
下一步是通过使用下拉手册将表面分配给数据集，将每个多物理场数据与相应的表面对齐。Zemax OpticStudio 中的 STAR 模块允许直接导入来自有限元分析（FEA）软件的变形数据，从而将光
阅读更多2025-01-21
拓扑学：单纯形(simplex)
拓扑学：单纯形(simplex)
阅读更多2025-01-21

多股票特征处理例子