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

fix syntax for Pandas dropping the name of field when is applied. (#6591)

Browse Source
This commit is contained in:
Danglewood
2024-07-18 03:20:10 -07:00
committed by GitHub
parent 2a967c6f49
commit 422d05077d
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
openbb_platform/extensions/fixedincome/openbb_fixedincome/fixedincome_views.py
View File

@@ -71,13 +71,14 @@ class FixedIncomeViews:
color = colors[color_count % len(colors)]
plot_df = df[df["date"] == date].copy()
plot_df["rate"] = plot_df["rate"].apply(lambda x: x * 100)
plot_df = plot_df.rename(columns={"rate": "Yield"})
plot_df = (
plot_df.drop(columns=["date"])
.set_index("maturity")
.filter(items=maturities, axis=0)
.reset_index()
.rename(columns={"maturity": "Maturity", "rate": "Yield"})
)
plot_df = plot_df.rename(columns={"index": "Maturity"})
plot_df["Maturity"] = [
(
d.split("_")[1] + " " + d.split("_")[0].title()
@@ -105,7 +106,7 @@ class FixedIncomeViews:
dates = df.date.unique().tolist()
figure, color_count = create_fig(figure, df, dates, color_count)
extra_params = kwargs.get("extra_params")
extra_params = kwargs.get("extra_params", {})
extra_params = (
extra_params if isinstance(extra_params, dict) else extra_params.__dict__
)