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)
sign0 = np.sign(cha0)
cha2 = np.diff(close)
sign2 = np.sign(cha2)
# 计算累加之和
csum = np.cumsum(sign2)
n = len(csum)
for i in range(1,n-2):
if (1==sign1[i]==sign1[i+1]==sign1[i+2]): # 连续3日阳线
if (1==sign0[i-1]==sign0[i]==sign0[i+1]): # 开盘价连升3日
if (1==sign2[i-1]==sign2[i]==sign2[i+1]): # 收盘价连升3日
if (csum[i-1] < csum[i] < csum[i+1]):
print('cumsum:',i,csum[i],csum[i+1],csum[i+2])
print("[date,open,close,high,low,volume,code]")
aline = df.loc[i-1, :].values # DataFrame 取一行数据
print(aline)
print(df.loc[i, :].values)
print(df.loc[i+1, :].values)
print(df.loc[i+2, :].values)
#
运行 python red_3.py 000061