wassname/pandas-ta
1
0
Fork 0
mirror of https://github.com/wassname/pandas-ta.git synced 2026-09-10 12:23:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'pr/121' into development

Browse Source
This commit is contained in:
Kevin Johnson
2020-09-10 13:34:57 -07:00
parent 339461ba8e 4a00ad79dd
commit 147872c2a4
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pandas_ta/core.py
+6 -3
View File
@@ -407,15 +407,18 @@ class AnalysisIndicators(BasePandasObject):
return getattr(self,method)(*args, **kwargs)[0]
def _post_process(self, result, **kwargs) -> (Series, DataFrame):
def _post_process(self, result, **kwargs) -> (pd.Series, pd.DataFrame):
"""Applies any additional modifications to the DataFrame
* Applies prefixes and/or suffixes
* Appends the result to main DataFrame
"""
if not isinstance(result, (Series, DataFrame)):
if not isinstance(result, (pd.Series, pd.DataFrame)):
print(f"[X] Oops! The result was not a Series or DataFrame.")
return self._df
else:
# Append only specific columns to the dataframe (via 'col_numbers':(0,1,3) for example)
result = result.iloc[:,[int(n) for n in kwargs['col_numbers']]] if isinstance(result, pd.DataFrame) and 'col_numbers' in kwargs and kwargs['col_numbers'] is not None else result
# Add prefix/suffix and append to the dataframe
self._add_prefix_suffix(result=result, **kwargs)
self._append(result=result, **kwargs)
return result
@@ -1472,4 +1475,4 @@ class AnalysisIndicators(BasePandasObject):
volume = self._get_column(kwargs.pop("volume", "volume"))
result = vp(close=close, volume=volume, width=width, percent=percent, **kwargs)
return self._post_process(result, **kwargs)
return self._post_process(result, **kwargs)