python：tushare 红三兵

最新推荐文章于 2024-11-20 09:03:47 发布

原创

最新推荐文章于 2024-11-20 09:03:47 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #tushare #红三兵

本文介绍了一种用于A股市场分析的红三兵算法，通过Python代码实现，该算法基于股票开盘价、收盘价、最高价和最低价进行计算，以识别连续三个交易日的阳线趋势，为投资者提供买入信号。

talib CDL3WHITESOLDIERS(open, high, low, close) 白三兵可能只适用于西方。

A股市股票红三兵算法： red_3.py

# coding: utf-8
import os, sys
import numpy as np
import pandas as pd
import tushare as ts
import math

if len(sys.argv) ==2:
    code = sys.argv[1]
    i = -1
elif len(sys.argv) ==3:
    code = sys.argv[1]
    i = int(sys.argv[2])
else:
    print('usage: python red_3.py stockcode ')
    sys.exit(1)

if len(code) !=6:
    print('stock code length: 6')
    sys.exit(2)

df = ts.get_k_data(code)
if df.empty ==True:
    print(" df is empty ")
    sys.exit(2)

df = df[ df['date'] > '2020-01-01']
df = df.reset_index(drop=True)
if len(df) <10:
    print(" len(df) <10 ")
    sys.exit(2)

# 开盘价
open = np.array(df['open'])
# 收盘价
close = np.array(df['close'])
# 最高价
high = np.array(df['high'])
# 最低价
low = np.array(df['low'])
#  收盘价 - 开盘价 = 差值
cha1 = close - open
sign1 = np.sign(cha1)
# diff函数可以计算数组中两个连续元素的差值，并返回一个由这些差值组成的数组：
cha0 = np.diff(open)
sign