python:talib 计算 KDJ

本文介绍如何使用Ta-lib库在Python中计算股票技术分析中的KDJ指标。通过示例代码展示了如何从tushare获取股票数据,利用Ta-lib计算KDJ值,并绘制KDJ曲线图与股票收盘价图。此教程适用于股票交易策略开发。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

用chrome 访问 https://www.lfd.uci.edu/~gohlke/pythonlibs/#ta-lib
下载 TA_Lib‑0.4.24‑cp37‑cp37m‑win_amd64.whl
pip install /pypi/TA_Lib‑0.4.24‑cp37‑cp37m‑win_amd64.whl

talib 计算 KDJ值对应的函数是Stochastic Oscillator Slow (Stoch),

其返回值有两个,一个是快速确认线值,另外一个是慢速主干线值。KDJ 需要至少最近9天的数据。

talib_kdj.py

# -*- coding: utf-8 -*-
import os, sys
import tushare as ts
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import talib

if len(sys.argv) ==2:
    code = sys.argv[1]
else:
    print('usage: python talib_kdj.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 :
    print(" df is empty ")
    sys.exit(2)

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

dw = pd.DataFrame()
# KDJ 值对应的函数是 STOCH
dw['slowk'], dw['slowd'] = talib.STOCH(
                        df['high'].values,
                        df['low'].values,
                        df['close'].values,
                        fastk_period=9,
                        slowk_period=3,
                        slowk_matype=0,
                        slowd_period=3,
                        slowd_matype=0)
# 求出J值,J = (3*K)-(2*D)
dw['slowj'] = list(map(lambda x,y: 3*x-2*y, dw['slowk'], dw['slowd']))
dw.index = range(len(dw))
print('df.len=',len(df),'dw.len=',len(dw))

df['ma10'] = df['close'].rolling(window=10).mean()
#print(df.head())
df = df.reset_index(drop=True)
# 取后半段
m = len(df) //2
df = df[ df.index > m ]
dw = dw[ dw.index > m ]
print(df.head())
print(dw.head())
df.index = pd.to_datetime(df.date)
dw.index = pd.to_datetime(df.date)
# 画股票收盘价图
fig,axes = plt.subplots(2,1)
df[['close', 'ma10']].plot(ax=axes[0], grid=True, title=code)
# 画 KDJ 曲线图
dw[['slowk','slowd','slowj']].plot(ax=axes[1], grid=True)
plt.legend(loc='best', shadow=True)
plt.show()

运行  python talib_kdj.py 600030

参考: 使用Ta-lib计算MACD与KDJ – 云原生之路

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值