python：tushare 红三兵

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