Skip to content

hal9000cc/pyita

Repository files navigation

pyita

PyPI - Version GitHub License PyPI - Python Version

pyita is a simple and fast technical analysis library for Python, written in pure Python with NumPy and Numba for maximum performance. It is ideal for Python-based financial market analysis, algorithmic trading, and working with data from any source.

Features

  • Pure native Python - written in clean Python, easy to read, understand, and modify
  • Blazingly fast - uses NumPy vectorized operations and Numba JIT compilation for maximum performance
  • Simple API - intuitive interface for working with OHLCV data
  • Rich set of indicators - 28 popular technical indicators
  • Flexible - easy to create custom indicators
  • Independent - no external data sources required, bring your own data from any source
  • Compatible - supports Python 3.9+ (tested up to 3.14)
  • Integrations - works with NumPy arrays, pandas DataFrames, and CCXT

What's New in 1.1.0

  • Added metadata() function to get metadata for all indicators (signatures, parameters, output series with types, descriptions)
  • Added list() function to get formatted list of all indicators in human-readable format
  • Added slicing support for DataSeries, Quotes, and IndicatorResult (returns views)
  • Fixed incorrect handling of CCXT format data in Quotes constructor

For full changelog, see CHANGELOG.md.

Installation

From PyPI:

pip install pyita

From source:

git clone https://github.com/hal9000cc/pyita.git
cd pyita
pip install -e .

Quick Start

From NumPy arrays:

import pyita as ta
import numpy as np

# Create quotes from arrays
quotes = ta.Quotes(
    open=np.array([100.0, 102.0, 101.0, 103.0]),
    high=np.array([105.0, 106.0, 104.0, 107.0]),
    low=np.array([99.0, 101.0, 100.0, 102.0]),
    close=np.array([102.0, 103.0, 101.0, 105.0])
)

# Calculate indicator
sma = ta.sma(quotes, period=3, value='close')
print(sma.sma)  # [nan, nan, 102.0, 103.0]

From pandas DataFrame:

import pandas as pd
import pyita as ta

# Load data from CSV
df = pd.read_csv('data.csv')
quotes = ta.Quotes(df)

# Calculate Bollinger Bands
bb = ta.bollinger_bands(quotes, period=20, deviation=2)
print(bb.up_line)    # Upper band
print(bb.mid_line)   # Middle line
print(bb.down_line)  # Lower band

From CCXT (cryptocurrency exchanges):

import ccxt
import pyita as ta

# Fetch data from exchange
exchange = ccxt.binance()
ohlcv = exchange.fetch_ohlcv('BTC/USDT', '1h', limit=100)
quotes = ta.Quotes(ohlcv)

# Calculate RSI
rsi = ta.rsi(quotes, period=14)
print(rsi.rsi)  # Values from 0 to 100

Creating Quotes Object

The Quotes object encapsulates OHLCV data and can be created from various sources.

Minimal variant (OHLC only):

import numpy as np
import pyita as ta

quotes = ta.Quotes(
    open=np.array([100.0, 102.0, 101.0]),
    high=np.array([105.0, 106.0, 104.0]),
    low=np.array([99.0, 101.0, 100.0]),
    close=np.array([102.0, 103.0, 101.0])
)

With volume:

quotes = ta.Quotes(
    open=open_prices,
    high=high_prices,
    low=low_prices,
    close=close_prices,
    volume=np.array([1000, 1200, 900])
)

With volume and time:

quotes = ta.Quotes(
    open=open_prices,
    high=high_prices,
    low=low_prices,
    close=close_prices,
    volume=volume,
    time=np.array(['2024-01-01', '2024-01-02', '2024-01-03'], dtype='datetime64[ms]')
)

From pandas DataFrame:

import pandas as pd

df = pd.DataFrame({
    'open': [100.0, 102.0],
    'high': [105.0, 106.0],
    'low': [99.0, 101.0],
    'close': [102.0, 103.0],
    'volume': [1000, 1200]
})

quotes = ta.Quotes(df)

From dictionary:

data = {
    'open': [100.0, 102.0],
    'high': [105.0, 106.0],
    'low': [99.0, 101.0],
    'close': [102.0, 103.0]
}

quotes = ta.Quotes(data)

From list of lists (CCXT format):

# CCXT returns: [[timestamp, open, high, low, close, volume], ...]
ohlcv = [
    [1609459200000, 100.0, 105.0, 99.0, 102.0, 1000],
    [1609545600000, 102.0, 106.0, 101.0, 103.0, 1200]
]

quotes = ta.Quotes(ohlcv)

Available Indicators

Moving Averages

  • sma(quotes, period, value='close') - Simple Moving Average

    • Requires: OHLC
    • Returns: sma
  • ema(quotes, period, value='close') - Exponential Moving Average

    • Requires: OHLC
    • Returns: ema
  • tema(quotes, period, value='close') - Triple Exponential Moving Average

    • Requires: OHLC
    • Returns: tema
  • vwma(quotes, period, value='close') - Volume Weighted Moving Average

    • Requires: OHLC + volume
    • Returns: vwma
  • ma(quotes, period, value='close', ma_type='sma') - Moving Average (universal)

Trend Indicators

  • adx(quotes, period=14, smooth=14, ma_type='mma') - Average Directional Index

    • Requires: OHLC
    • Returns: adx, p_di (Plus DI), m_di (Minus DI)
  • aroon(quotes, period=14) - Aroon Indicator

    • Requires: OHLC (uses high, low)
    • Returns: up, down, oscillator
  • parabolic_sar(quotes, start=0.02, maximum=0.2, increment=0.02) - Parabolic SAR

    • Requires: OHLC
    • Returns: sar, signal
  • supertrend(quotes, period=10, multiplier=3, ma_type='mma') - SuperTrend

    • Requires: OHLC
    • Returns: supertrend, signal
  • macd(quotes, period_fast=12, period_slow=26, period_signal=9, value='close') - Moving Average Convergence Divergence

    • Requires: OHLC
    • Returns: macd, signal, histogram
  • ichimoku(quotes, period_short=9, period_mid=26, period_long=52, offset_senkou=26, offset_chikou=26) - Ichimoku Cloud

    • Requires: OHLC (uses high, low)
    • Returns: tenkan, kijun, senkou_a, senkou_b, chikou

Oscillators

  • rsi(quotes, period=14, ma_type='mma', value='close') - Relative Strength Index

    • Requires: OHLC
    • Returns: rsi (values from 0 to 100)
  • stochastic(quotes, period=5, period_d=3, smooth=3, ma_type='sma') - Stochastic Oscillator

    • Requires: OHLC
    • Returns: value_k, value_d, oscillator
  • williams_r(quotes, period=14) - Williams %R

    • Requires: OHLC
    • Returns: williams_r (values from -100 to 0)
  • cci(quotes, period=20) - Commodity Channel Index

    • Requires: OHLC
    • Returns: cci
  • mfi(quotes, period=14) - Money Flow Index

    • Requires: OHLC + volume
    • Returns: mfi
  • roc(quotes, period=14, value='close') - Rate of Change

    • Requires: OHLC
    • Returns: roc
  • awesome(quotes, period_fast=5, period_slow=34, normalized=False) - Awesome Oscillator

    • Requires: OHLC (uses high, low)
    • Returns: awesome
  • trix(quotes, period, value='close') - Triple Exponential Average Oscillator

    • Requires: OHLC
    • Returns: trix

Volatility

  • bollinger_bands(quotes, period=20, deviation=2, ma_type='sma', value='close') - Bollinger Bands

    • Requires: OHLC
    • Returns: mid_line, up_line, down_line, width, z_score
  • atr(quotes, smooth=14, ma_type='mma') - Average True Range

    • Requires: OHLC
    • Returns: atr, atrp (percentage ATR), tr (True Range)
  • keltner(quotes, period=10, multiplier=1, period_atr=10, ma_type='ema') - Keltner Channels

    • Requires: OHLC
    • Returns: mid_line, up_line, down_line, width
  • chandelier(quotes, period=22, multiplier=3, use_close=False) - Chandelier Exit

    • Requires: OHLC
    • Returns: exit_long, exit_short

Volume Indicators

  • obv(quotes) - On-Balance Volume

    • Requires: OHLC + volume (uses close, volume)
    • Returns: obv
  • vwap(quotes) - Volume Weighted Average Price

    • Requires: OHLC + volume
    • Returns: vwap
  • volume_osc(quotes, period_short=5, period_long=10, ma_type='ema') - Volume Oscillator

    • Requires: volume
    • Returns: osc (percentage difference between short and long MA of volume)
  • adl(quotes, ma_period=None, ma_type='sma') - Accumulation/Distribution Line

    • Requires: OHLC + volume
    • Returns: adl, adl_ema (if ma_period is specified)

Other Indicators

  • zigzag(quotes, delta=0.02, depth=1, type='high_low', end_points=False) - ZigZag
    • Requires: OHLC
    • Returns: pivots, pivot_types (1 = High, -1 = Low, 0 = no pivot)

Moving Average Types

Many indicators support selecting the moving average type via the ma_type parameter:

  • sma - Simple Moving Average

    • All values have equal weight
    • Initialization: SMA over first period values
  • ema - Exponential Moving Average

    • More weight given to recent values
    • Smoothing coefficient: α = 2 / (period + 1)
    • Initialization: SMA over first period values
  • mma (or smma, rma) - Modified/Smoothed Moving Average

    • Similar to EMA but with slower response
    • Smoothing coefficient: α = 1 / period
    • Initialization: SMA over first period values
    • Used in RSI, ADX indicators
  • ema0 - EMA with first value initialization

    • No warmup period required
  • mma0 - MMA with first value initialization

    • No warmup period required
  • emaw - EMA with dynamic warmup period (TA-Lib compatible)

  • mmaw - MMA with dynamic warmup period (TA-Lib compatible)

Essentially, technical analysis uses only two moving average algorithms - simple (sma) and exponential (ema). SMA is always calculated the same way. EMA is also calculated consistently, but there can be variations in how it uses the period (smoothing coefficient calculation) and how it's initialized. This is what distinguishes the different ema types supported by pyita.

Usage example:

# Bollinger Bands with EMA
bb = ta.bollinger_bands(quotes, period=20, deviation=2, ma_type='ema')

# Keltner Channels with MMA
keltner = ta.keltner(quotes, period=20, multiplier=2, period_atr=10, ma_type='mma')

# RSI uses MMA by default (can be changed)
rsi = ta.rsi(quotes, period=14, ma_type='ema')

Metadata

The library provides functions to get metadata about all available indicators.

metadata() - Returns a dictionary with metadata for all indicators:

import pyita as ta

meta = ta.metadata()
print(meta['bollinger_bands'])

Output:

{
    'name': 'bollinger_bands',
    'signature': 'bollinger_bands(quotes, period=20, deviation=2, ma_type=\'sma\', value=\'close\')',
    'parameters': ['quotes', 'period', 'deviation', 'ma_type', 'value'],
    'output_series': [
        {'name': 'mid_line', 'type': 'price'},
        {'name': 'up_line', 'type': 'price'},
        {'name': 'down_line', 'type': 'price'},
        {'name': 'z_score', 'type': 'none'}
    ],
    'description': 'Bollinger bands'
}

list() - Returns a formatted string with all indicators in human-readable format:

import pyita as ta

print(ta.list())

Output:

adl(quotes, ma_period=None, ma_type='sma')
  Accumulation/distribution line.
  Output: adl, adl_smooth

adx(quotes, period=14, smooth=14, ma_type='mma')
  Average directional movement index.
  Output: adx, p_di, m_di

bollinger_bands(quotes, period=20, deviation=2, ma_type='sma', value='close')
  Bollinger bands.
  Output: mid_line (price), up_line (price), down_line (price), z_score

ema(quotes, period, value='close')
  Exponential moving average.
  Output: ema (as source)

...

Series Types:

In the output series, types are indicated in parentheses:

  • (price) - Price-based series that are displayed on the price chart (e.g., moving averages, channel lines)
  • (as source) - Series whose type depends on the source data (e.g., ma can be calculated on price or volume)
  • No type - Value-based series displayed on separate charts (e.g., oscillators, indices, signals)

System Requirements

  • Python: 3.9+ (tested up to 3.14)
  • Required dependencies:
    • numpy >= 1.20.0
    • numba >= 0.53.0
  • Optional dependencies:
    • pandas >= 1.3.0 (for DataFrame support)

Testing

# Install development dependencies
pip install -e ".[dev]"

# Run tests
pytest tests/

License

MIT License

Copyright (c) 2026 Aleksandr Kuznetsov hal@hal9000.cc

Contact

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors