jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 13:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add to_dict method

Browse Source
This commit is contained in:
Diogo Sousa
2023-07-05 18:21:34 +01:00
parent e5f2438a6b
commit d79da5fae1
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/model/command_output.py
View File

@@ -52,7 +52,7 @@ class CommandOutput(GenericModel, Generic[T], Tagged):
return df
def to_lists(self, fields: Optional[List[str]] = None) -> Dict[str, List]:
def to_dict(self) -> Dict[str, List]:
"""Converts results field to list of values.
Returns
@@ -61,11 +61,8 @@ class CommandOutput(GenericModel, Generic[T], Tagged):
Dictionary of lists.
"""
df = self.to_dataframe()
target_fields = df.columns if fields is None else fields
results = {}
for field in target_fields:
if field not in df.columns:
choices = ", ".join(df.columns)
raise ValueError(f"Field {field} not found, choose from: {choices}.")
for field in df.columns:
results[field] = df[field].tolist()
return results