在backtrader中,可以通过定义自定义列来创建指标。在此过程中,在DataFrame中定义自定义列是必要的。backtrader并不会自动调用这些列以生成指标,而必须在策略中手动指定指标。
以下是一个示例,以添加一个自定义指标"my_indicator"为例:
import backtrader as bt
class MyStrategy(bt.Strategy):
def __init__(self):
self.my_indicator = self.data.close * 2
def next(self):
print('My Indicator:', self.my_indicator[0])
cerebro = bt.Cerebro()
data = bt.feeds.YahooFinanceData(dataname='AAPL', fromdate=datetime(2020, 1, 1), todate=datetime(2020, 12, 31))
cerebro.adddata(data)
cerebro.addstrategy(MyStrategy)
cerebro.run()
在上面的代码中,我们定义了一个自定义指标"my_indicator",它是data的"close"列的两倍。我们使用策略类中的"init()"函数来定义自定义指标,然后在"next()"函数中打印出自定义指标的值。
当我们运行 cerebro.run() 时,我们能获得以下输出:
My Indicator: 296.03692626953125
My Indicator: 301.5375671386719
My Indicator: 301.9408874511719
My Indicator: 298.2789611816406
My Indicator: 298.9488220214844
My Indicator: 295.5012969970703
...
由此可见,我们已经成功添加并输出自定义指标"my_indicator"。
图示代码:
import backtrader as bt
import pandas as pd
class MyStrategy(bt.Strategy):
params = (('n', 20),)
def __init__(self):
self.my_indicator = self.data.close * 2
def next(self):
print('My Indicator:', self.my_indicator[0])