首页 > 解决方案 > 带有热图的 Matplotlib 用于在后台进行预测

问题描述

我尝试了以下方法:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
df = pd.read_csv("t.csv")
fig = plt.figure()
fig.set_size_inches(50,70)
ax1 = fig.add_subplot(111)
ax1.scatter(np.arange(len(df.index)),df['real'].values,c=df["colors_real"].values, alpha=0.9)
ax2 = ax1.twinx()
ax2.hist2d(np.arange(len(df.index)),df['prediction'].values, bins=30, cmap=plt.cm.jet,alpha=0.5)
plt.show()

上图所需的数据集:t.csv
我得到了下图: 但我期望这个图是这样的:
输出图

预期图像

我想要:
1)用df['real']值制作圆圈,其中负值圆圈的中心是石灰色,然后是0的值,然后是预期数字中所有值的正值。
2)放置热图或hist2d相对于值,石灰色为负,橙色为正。
请让我知道我该怎么做。我想按预期的数字显示输出。

标签: pythonpython-3.xmatplotlibvisualization

解决方案


推荐阅读