Skip to content

indicator ¤

above ¤

above(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is above another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: True series1 is above series2 else False

Source code in lettrade/indicator/series.py
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
def above(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is above another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: True series1 is above series2 else False
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    i = s_above(series1=series1, series2=series2)

    if inplace:
        name = name or f"{prefix}above"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

below ¤

below(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is below another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: -100 series1 is below series2 else 0

Source code in lettrade/indicator/series.py
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
def below(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is below another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: -100 series1 is below series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    i = s_below(series1, series2)

    if inplace:
        name = name or f"{prefix}below"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

cdl_3blackcrows ¤

cdl_3blackcrows(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
def cdl_3blackcrows(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="3blackcrows",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_3inside ¤

cdl_3inside(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
def cdl_3inside(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="3inside",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_3whitesoldiers ¤

cdl_3whitesoldiers(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
def cdl_3whitesoldiers(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="3whitesoldiers",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_direction ¤

cdl_direction(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
) -> Series | DataFrame

Direction of candle

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: pd.Series return [100, 0 , -100] - 100 for bull bar - -100 for bear bar - 0 for None

Source code in lettrade/indicator/candlestick/candlestick.py
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def cdl_direction(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
) -> pd.Series | pd.DataFrame:
    """Direction of candle

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: `pd.Series` return [100, 0 , -100]
            - `100` for bull bar
            - `-100` for bear bar
            - `0` for None
    """
    if __debug__:
        if not isinstance(dataframe, pd.DataFrame):
            raise RuntimeError(
                f"dataframe type '{type(dataframe)}' "
                "is not instance of pandas.DataFrame"
            )
    i = dataframe.apply(
        lambda r: 100 if r.open < r.close else -100 if r.open > r.close else 0,
        axis=1,
    ).astype(int)

    if inplace:
        dataframe[name or f"{prefix}direction"] = i
        return dataframe

    return i

cdl_doji ¤

cdl_doji(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
def cdl_doji(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="doji",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_engulfing ¤

cdl_engulfing(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
def cdl_engulfing(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="engulfing",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_eveningstar ¤

cdl_eveningstar(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
def cdl_eveningstar(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="eveningstar",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_morningstar ¤

cdl_morningstar(
    dataframe: DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    description

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
def cdl_morningstar(
    dataframe: pd.DataFrame,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): _description_
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return cdl_pattern(
        dataframe=dataframe,
        pattern="morningstar",
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

cdl_pattern ¤

cdl_pattern(
    dataframe: DataFrame,
    pattern: str,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal[
        "candlestick", "mark"
    ] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

summary

Parameters:

  • dataframe (DataFrame) –

    pandas.DataFrame with ohlcv

  • pattern (str) –

    TA-Lib candle pattern name. Ex: 3whitesoldiers, 3blackcrows

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: 'cdl_' ) –

    description. Defaults to "cdl_".

  • inplace (bool, default: False ) –

    description. Defaults to False.

Raises:

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/candlestick/talib.py
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
def cdl_pattern(
    dataframe: pd.DataFrame,
    pattern: str,
    name: str | None = None,
    prefix: str = "cdl_",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["candlestick", "mark"] = "candlestick",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """_summary_

    Args:
        dataframe (pd.DataFrame): pandas.DataFrame with ohlcv
        pattern (str): TA-Lib candle pattern name. Ex: `3whitesoldiers`, `3blackcrows`
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "cdl_".
        inplace (bool, optional): _description_. Defaults to False.

    Raises:
        RuntimeError: _description_

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    if __debug__:
        if not isinstance(dataframe, pd.DataFrame):
            raise RuntimeError(
                f"dataframe type '{type(dataframe)}' "
                "is not instance of pandas.DataFrame"
            )
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    i = getattr(ta, f"CDL{pattern.upper()}")(dataframe, **kwargs)

    if inplace:
        name = name or f"{prefix}{pattern.lower()}"
        dataframe[name] = i

        # Plot
        if plot:
            _plot_pattern(
                dataframe=dataframe,
                indicator=i,
                name=name,
                plot_type=plot_type,
                plot_kwargs=plot_kwargs,
            )

        return dataframe

    return i

crossover ¤

crossover(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check if a Series cross over another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: 100 if series1 cross over series2 else 0

Source code in lettrade/indicator/series.py
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
def crossover(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check if a Series cross over another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: 100 if series1 cross over series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    below1 = s_below(series1, series2).shift(1)
    above0 = s_above(series1, series2)
    i = (-below1 + above0).apply(lambda v: 100 if v >= 200 else 0)

    if inplace:
        name = name or f"{prefix}crossover"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

crossunder ¤

crossunder(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check if a Series cross under another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: -100 if series1 cross under series2 else 0

Source code in lettrade/indicator/series.py
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
def crossunder(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check if a Series cross under another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: -100 if series1 cross under series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    above1 = s_above(series1, series2).shift(1)
    below0 = s_below(series1, series2)
    i = (below0 - above1).apply(lambda v: -100 if v <= -200 else 0)

    if inplace:
        name = name or f"{prefix}crossunder"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

diff ¤

diff(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Difference between 2 series

Parameters:

  • series1 (Series) –

    description

  • series2 (Series) –

    description

  • dataframe (DataFrame, default: None ) –

    description. Defaults to None.

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: '' ) –

    description. Defaults to "".

  • inplace (bool, default: False ) –

    description. Defaults to False.

  • plot (bool | list, default: False ) –

    description. Defaults to False.

  • plot_kwargs (dict | None, default: None ) –

    description. Defaults to None.

Raises:

Returns:

  • Series

    pd.Series: Difference of 2 series (series1 - series2)

Source code in lettrade/indicator/series.py
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
def diff(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Difference between 2 series

    Args:
        series1 (pd.Series): _description_
        series2 (pd.Series): _description_
        dataframe (pd.DataFrame, optional): _description_. Defaults to None.
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "".
        inplace (bool, optional): _description_. Defaults to False.
        plot (bool | list, optional): _description_. Defaults to False.
        plot_kwargs (dict | None, optional): _description_. Defaults to None.

    Raises:
        RuntimeError: _description_

    Returns:
        pd.Series: Difference of 2 series (series1 - series2)
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    i = series1 - series2

    if inplace:
        name = name or f"{prefix}diff"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

direction ¤

direction(
    series1: Series | str,
    series2: Series | str,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is direction another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: -100 if series1 < series2, 100 if series1 > series2, else 0

Source code in lettrade/indicator/series.py
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
def direction(
    series1: pd.Series | str,
    series2: pd.Series | str,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is direction another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: -100 if series1 < series2, 100 if series1 > series2, else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    i = s_direction(series1, series2)

    if inplace:
        name = name or f"{prefix}direction"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

ema ¤

ema(
    series: Series | str = "close",
    window: int = None,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

Exponential Moving Average

Parameters:

  • series (Series | str, default: 'close' ) –

    description. Defaults to "close".

  • window (int, default: None ) –

    description. Defaults to None.

  • dataframe (DataFrame, default: None ) –

    description. Defaults to None.

  • name (str | None, default: None ) –

    description. Defaults to None.

  • prefix (str, default: '' ) –

    description. Defaults to "".

  • inplace (bool, default: False ) –

    description. Defaults to False.

  • plot (bool | list, default: False ) –

    description. Defaults to False.

  • plot_kwargs (dict | None, default: None ) –

    description. Defaults to None.

Example
df.i.ema(window=21, name="ema", inplace=True, plot=True)

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/trend/ma.py
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
def ema(
    series: pd.Series | str = "close",
    window: int = None,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """Exponential Moving Average

    Args:
        series (pd.Series | str, optional): _description_. Defaults to "close".
        window (int, optional): _description_. Defaults to None.
        dataframe (pd.DataFrame, optional): _description_. Defaults to None.
        name (str | None, optional): _description_. Defaults to None.
        prefix (str, optional): _description_. Defaults to "".
        inplace (bool, optional): _description_. Defaults to False.
        plot (bool | list, optional): _description_. Defaults to False.
        plot_kwargs (dict | None, optional): _description_. Defaults to None.

    Example:
        ```python
        df.i.ema(window=21, name="ema", inplace=True, plot=True)
        ```

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return ma(
        series=series,
        window=window,
        mode="ema",
        dataframe=dataframe,
        name=name,
        prefix=prefix,
        inplace=inplace,
        plot=plot,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )

indicators_inject_pandas ¤

indicators_inject_pandas(obj: object | None = None)

Inject indicators to Pandas

Source code in lettrade/indicator/__init__.py
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
def indicators_inject_pandas(obj: object | None = None):
    """Inject indicators to Pandas"""
    if obj is None:
        from pandas.core.base import PandasObject

        obj = PandasObject

    # Flag to mark indicators injected
    if hasattr(obj, "_lt_indicators_injected"):
        return

    from .candlestick import pandas_inject as candlestick_pandas_inject
    from .dataframe import pandas_inject as dataframe_pandas_inject
    from .momentum import pandas_inject as momentum_pandas_inject
    from .series import pandas_inject as series_pandas_inject
    from .trend import pandas_inject as trend_pandas_inject
    from .volatility import pandas_inject as volatility_pandas_inject

    series_pandas_inject(obj)
    dataframe_pandas_inject(obj)

    candlestick_pandas_inject(obj)
    trend_pandas_inject(obj)
    volatility_pandas_inject(obj)
    momentum_pandas_inject(obj)

    # Flag to mark indicators injected
    obj._lt_indicators_injected = True

rolling_above ¤

rolling_above(
    series1: Series,
    series2: Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is rolling above another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: 100 mean series1 is rolling above series2 else 0

Source code in lettrade/indicator/series.py
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
def rolling_above(
    series1: pd.Series,
    series2: pd.Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is rolling above another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: 100 mean series1 is rolling above series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    min_periods = window if min_periods is None else min_periods

    i = s_above(series1, series2)
    i = i.rolling(window=window, min_periods=min_periods).min()
    i = i.apply(lambda v: 100 if v >= 100 else 0).astype(int)

    if inplace:
        name = name or f"{prefix}rolling_above"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

rolling_below ¤

rolling_below(
    series1: Series,
    series2: Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is rolling below another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: -100 mean series1 is rolling below series2 else 0

Source code in lettrade/indicator/series.py
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
def rolling_below(
    series1: pd.Series,
    series2: pd.Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is rolling below another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: -100 mean series1 is rolling below series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    min_periods = window if min_periods is None else min_periods

    i = s_below(series1, series2)
    i = i.rolling(window=window, min_periods=min_periods).max()
    i = i.apply(lambda v: -100 if v <= -100 else 0).astype(int)

    if inplace:
        name = name or f"{prefix}rolling_below"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

rolling_direction ¤

rolling_direction(
    series1: Series,
    series2: Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
) -> Series

Check a Series is rolling on one side with another Series

Parameters:

  • series1 (Series) –

    first Series

  • series2 (Series) –

    second Series

Returns:

  • Series

    pd.Series: -100 mean series1 is rolling direction series2 else 0

Source code in lettrade/indicator/series.py
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
def rolling_direction(
    series1: pd.Series,
    series2: pd.Series,
    window: int = 20,
    min_periods: int | None = None,
    dataframe: pd.DataFrame = None,
    name: str | None = None,
    prefix: str = "",
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_kwargs: dict | None = None,
    # **kwargs,
) -> pd.Series:
    """Check a Series is rolling on one side with another Series

    Args:
        series1 (pd.Series): first Series
        series2 (pd.Series): second Series

    Returns:
        pd.Series: -100 mean series1 is rolling direction series2 else 0
    """
    if __debug__:
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    if isinstance(series1, str):
        series1 = dataframe[series1]
    if isinstance(series2, str):
        series2 = dataframe[series2]

    min_periods = window if min_periods is None else min_periods

    i = s_direction(series1, series2)
    i = i.rolling(window=window, min_periods=min_periods).mean()
    i = i.apply(lambda v: -100 if v <= -100 else 100 if v >= 100 else 0).astype(int)

    if inplace:
        name = name or f"{prefix}rolling_direction"
        dataframe[name] = i

        if plot:
            _plot_mark(dataframe=dataframe, name=name, plot_kwargs=plot_kwargs)

    return i

signal_condiction ¤

signal_condiction(
    dataframe: DataFrame,
    *condictions: list[list[Series | Any]],
    name: str,
    value: int | float = np.nan,
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["line", "mark"] = "line",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

Define a signal with multiple condiction

Parameters:

  • dataframe (DataFrame) –

    description

  • *condictions (list[list[Series | Any]], default: () ) –

    Pairs of condiction [<pandas.Series condiction>, <value>]

  • name (str) –

    Name of signal, column name when add to DataFrame with inplace=True.

  • value (int, default: nan ) –

    Default value when condiction is not matched. Defaults to 0.

  • inplace (bool, default: False ) –

    description. Defaults to False.

  • plot (bool | list, default: False ) –

    description. Defaults to False.

  • plot_type (Literal[&quot;line&quot;, &quot;mark&quot;], default: 'line' ) –

    description. Defaults to "line".

  • plot_kwargs (dict | None, default: None ) –

    description. Defaults to None.

Usage
df.i.signal_condiction(
    [df["close"] > df["open"], 100],
    [df["close"] < df["open"], -100],
    name="cdl_direction",
    inplace=True,
    plot=True,
    plot_kwargs=dict(color="green", width=5),
)

Raises:

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/dataframe.py
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
def signal_condiction(
    dataframe: pd.DataFrame,
    *condictions: list[list[pd.Series | Any]],
    name: str,
    value: int | float = np.nan,
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["line", "mark"] = "line",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """Define a signal with multiple condiction

    Args:
        dataframe (pd.DataFrame): _description_
        *condictions (list[list[pd.Series | Any]]): Pairs of condiction [`<pandas.Series condiction>`, `<value>`]
        name (str): Name of signal, column name when add to DataFrame with inplace=True.
        value (int, optional): Default value when condiction is not matched. Defaults to 0.
        inplace (bool, optional): _description_. Defaults to False.
        plot (bool | list, optional): _description_. Defaults to False.
        plot_type (Literal[&quot;line&quot;, &quot;mark&quot;], optional): _description_. Defaults to "line".
        plot_kwargs (dict | None, optional): _description_. Defaults to None.

    Usage:
        ```python
        df.i.signal_condiction(
            [df["close"] > df["open"], 100],
            [df["close"] < df["open"], -100],
            name="cdl_direction",
            inplace=True,
            plot=True,
            plot_kwargs=dict(color="green", width=5),
        )
        ```

    Raises:
        RuntimeError: _description_

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    if __debug__:
        if not isinstance(dataframe, pd.DataFrame):
            raise RuntimeError(
                f"dataframe type '{type(dataframe)}' "
                "is not instance of pandas.DataFrame"
            )
        if plot and not inplace:
            raise RuntimeError("Cannot plot when inplace=False")

    s = pd.Series(value, index=dataframe.index, name=name, **kwargs)
    for condiction in condictions:
        s.loc[condiction[0]] = condiction[1]

    if inplace:
        dataframe[name] = s

        # Plot
        if plot:
            if plot_kwargs is None:
                plot_kwargs = dict()

            plot_kwargs.update(series=name, name=name)

            from lettrade.indicator.plot import IndicatorPlotter
            from lettrade.plot.plotly import plot_line, plot_mark

            plotter = plot_mark if plot_type == "mark" else plot_line
            IndicatorPlotter(dataframe=dataframe, plotter=plotter, **plot_kwargs)

        return dataframe

    return s

signal_direction ¤

signal_direction(
    dataframe: DataFrame,
    up: Series,
    down: Series,
    name: str,
    value: int = 0,
    value_up: int = 100,
    value_down: int = -100,
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["line", "mark"] = "line",
    plot_kwargs: dict | None = None,
    **kwargs
) -> Series | DataFrame

Define a signal with 2 direction Up and Down with fixed value

Parameters:

  • dataframe (DataFrame) –

    description

  • up (Series) –

    description

  • down (Series) –

    description

  • name (str) –

    Name of signal, column name when add to DataFrame with inplace=True.

  • value (int, default: 0 ) –

    Default value when condiction is not matched. Defaults to 0.

  • value_up (int, default: 100 ) –

    description. Defaults to 100.

  • value_down (int, default: -100 ) –

    description. Defaults to -100.

  • inplace (bool, default: False ) –

    Whether to add to the DataFrame and return DataFrame rather than return result. Defaults to False.

  • plot (bool | list, default: False ) –

    description. Defaults to False.

  • plot_type (Literal[&quot;line&quot;, &quot;mark&quot;], default: 'line' ) –

    description. Defaults to "line".

  • plot_kwargs (dict | None, default: None ) –

    description. Defaults to None.

Returns:

  • Series | DataFrame

    pd.Series | pd.DataFrame: description

Source code in lettrade/indicator/dataframe.py
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
def signal_direction(
    dataframe: pd.DataFrame,
    up: pd.Series,
    down: pd.Series,
    name: str,
    value: int = 0,
    value_up: int = 100,
    value_down: int = -100,
    inplace: bool = False,
    plot: bool | list[str] = False,
    plot_type: Literal["line", "mark"] = "line",
    plot_kwargs: dict | None = None,
    **kwargs,
) -> pd.Series | pd.DataFrame:
    """Define a signal with 2 direction Up and Down with fixed value

    Args:
        dataframe (pd.DataFrame): _description_
        up (pd.Series): _description_
        down (pd.Series): _description_
        name (str): Name of signal, column name when add to DataFrame with inplace=True.
        value (int, optional): Default value when condiction is not matched. Defaults to 0.
        value_up (int, optional): _description_. Defaults to 100.
        value_down (int, optional): _description_. Defaults to -100.
        inplace (bool, optional): Whether to add to the DataFrame and return DataFrame rather than return result. Defaults to False.
        plot (bool | list, optional): _description_. Defaults to False.
        plot_type (Literal[&quot;line&quot;, &quot;mark&quot;], optional): _description_. Defaults to "line".
        plot_kwargs (dict | None, optional): _description_. Defaults to None.

    Returns:
        pd.Series | pd.DataFrame: _description_
    """
    return signal_condiction(
        dataframe,
        [up, value_up],
        [down, value_down],
        name=name,
        value=value,
        inplace=inplace,
        plot=plot,
        plot_type=plot_type,
        plot_kwargs=plot_kwargs,
        **kwargs,
    )