在 3-d 散点图中画线 plot

Question

我有一个 3d 散点图 plot。我想画两条线来描述 2021 年和 2022 年的点。有什么办法可以在 matplotlib 中做到这一点？ 我会附上我希望它看起来像。 黑线是我需要的。

我试着问 chat gpt，但它不明白我想做什么。

Answer 1

我已经解决了。 如果有人感兴趣，这是一个代码：

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.scatter(playlists["Year"], playlists["Playlist title number"] , playlists["Playlist views"], c='r', marker='o')
ax.set_xticks([2021, 2022])

ax.set_xlabel('Year')
ax.set_ylabel('Playlist Title Number')
ax.set_zlabel('Views')

playlists_2021=playlists.loc[playlists['Year'] == 2021]
playlists_2022=playlists.loc[playlists['Year'] == 2022]

x = playlists_2021["Playlist title number"].values.reshape(-1, 1)
y = playlists_2021["Playlist views"]
reg = LinearRegression().fit(x, y)

y_pred = reg.predict(x)

ax.plot(np.tile(2021, y_pred.shape), playlists_2021["Playlist title number"], y_pred, color='blue', linewidth=2)

x = playlists_2022["Playlist title number"].values.reshape(-1, 1)
y = playlists_2022["Playlist views"]
reg = LinearRegression().fit(x, y)

y_pred = reg.predict(x)

ax.plot(np.tile(2022, y_pred.shape), playlists_2022["Playlist title number"], y_pred, color='blue', linewidth=2)


plt.show()