Отсутствующая линия тренда - PullRequest
0 голосов
/ 07 мая 2018

Я пытаюсь добавить линию тренда к историческим значениям S & P500, но когда я пытаюсь добавить линию тренда, она не появляется, на графике отображаются только значения S & P500. («Дато» означает «Дата» на норвежском языке.)

Вот мой код:

##################### Load the data

df = pd.read_csv("SP500.csv", delimiter=r",", names = ["Date","Open", "High",
                                      "Low", "Close", "S&P500",
                                      "Volume"])

df = df.drop(columns=['Open', 'High', 'Low', 'Close', 'Volume'])

df["Dato"] = df["Date"]

df2 = pd.read_excel("OSEBX.xlsx", names = ["Date", "OSEBX"])

df.set_index('Date', inplace=True)
df2.set_index('Date', inplace=True)

df3 = df.join(df2)
df["Dato"] = pd.to_datetime(df3["Dato"])

################################ Graph


t = np.array(range(0, len(df["Dato"])))

x = np.array(df3["OSEBX"])
y = np.array(df3["S&P500"])

plt.scatter(t,x, s=1)

z = np.polyfit(x, t, 1)

p = np.poly1d(z)

plt.plot(x,p(x),"bs", linewidth=20)


plt.show()

1 Ответ

0 голосов
/ 07 мая 2018
#Thanks to Andreas Deak I found a solution:

t = np.array(range(0, len(df3["Dato"])))

x = np.array(df3["OSEBX"])
y = np.array(df3["S&P500"])

plt.plot(t, x, linewidth=0.5, color="b")

z = np.polyfit(t, x, 1)
p = np.poly1d(z)
plt.plot(t, p(t),"r--", linewidth=0.5)
plt.show()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...