mirror of
https://github.com/wassname/pandas-ta.git
synced 2026-08-16 11:25:01 +08:00
ENH #163 Adding Price Volume indicator
This commit is contained in:
@@ -1547,6 +1547,13 @@ class AnalysisIndicators(BasePandasObject):
|
||||
result = pvol(close=close, volume=volume, offset=offset, **kwargs)
|
||||
return self._post_process(result, **kwargs)
|
||||
|
||||
def pvr(self, **kwargs):
|
||||
close = self._get_column(kwargs.pop("close", "close"))
|
||||
volume = self._get_column(kwargs.pop("volume", "volume"))
|
||||
|
||||
result = pvr(close=close, volume=volume)
|
||||
return self._post_process(result, **kwargs)
|
||||
|
||||
def pvt(self, offset=None, **kwargs):
|
||||
close = self._get_column(kwargs.pop("close", "close"))
|
||||
volume = self._get_column(kwargs.pop("volume", "volume"))
|
||||
|
||||
@@ -10,5 +10,6 @@ from .nvi import nvi
|
||||
from .obv import obv
|
||||
from .pvi import pvi
|
||||
from .pvol import pvol
|
||||
from .pvr import pvr
|
||||
from .pvt import pvt
|
||||
from .vp import vp
|
||||
|
||||
@@ -134,6 +134,16 @@ class TestVolume(TestCase):
|
||||
self.assertIsInstance(result, Series)
|
||||
self.assertEqual(result.name, "PVOL")
|
||||
|
||||
def test_pvr(self):
|
||||
result = pandas_ta.pvr(self.close, self.volume_)
|
||||
self.assertIsInstance(result, Series)
|
||||
self.assertEqual(result.name, "PVR")
|
||||
# sample indicator values from SPY
|
||||
self.assertEqual(result[0], 1)
|
||||
self.assertEqual(result[1], 3)
|
||||
self.assertEqual(result[4], 2)
|
||||
self.assertEqual(result[6], 4)
|
||||
|
||||
def test_pvt(self):
|
||||
result = pandas_ta.pvt(self.close, self.volume_)
|
||||
self.assertIsInstance(result, Series)
|
||||
|
||||
Reference in New Issue
Block a user