如何在Plotnine中绘制2维密度 /热图？

Question

我正在尝试在plotnine 中重新创建以下图表。它要求我提供更多细节，但我不想分散对示例的注意力。我认为我想做的事情很明显。一位同事赋予了我一项职能。我对重写该函数不感兴趣。我想使用 sm 并使用plotnine而不是matplotlib来绘制它。我用plotnine绘制了很多数据帧，但我不确定在这种情况下如何使用它。我曾尝试自己解决这个问题，但我总是迷失方向。我希望对于更有经验的人来说，我忽略了一些简单的事情。

import matplotlib.pyplot as plt

def getSuccess(y,x):
    return((y*(-x))*.5+.5)

steps = 100
stepSize = 1/steps

sm = []

for y in range(steps*2+1):
    sm.append([getSuccess((y-steps)*stepSize,(x-steps)*stepSize) for x in range(steps*2+1)])

plt.imshow(sm)

plt.ylim(-1, 1)

plt.colorbar()
plt.yticks([0,steps,steps*2],[str(y) for y in [-1.0,0.0,1.0]])
plt.xticks([0,steps,steps*2],[str(x) for x in [-1.0,0.0,1.0]])
plt.show()

Answer 1

您可以尝试geom_raster。

我已获取您的合成数据 sm 并将其转换为 dataframe，因为 plotnine 需要它。

import pandas as pd
import numpy as np
from plotnine import *

df = pd.DataFrame(sm).melt()
df.rename(columns={'variable':'x','value':'density'}, inplace=True)
df.insert(1,'y',df.index % 201)

p = (ggplot(df, aes('x','y'))
  + geom_raster(aes(fill='density'), interpolate=True)
  + labs(x=None,y=None)
  + scale_x_continuous(expand=(0,0), breaks=[0,100,200], labels=[-1,0,1])
  + scale_y_continuous(expand=(0,0), breaks=[0,100,200], labels=[-1,0,1])
  + theme_matplotlib()
  + theme(
      text = element_text(family="Calibri", size=9),
      legend_title = element_blank(),
      axis_ticks = element_blank(),
      legend_key_height = 29.6,
      legend_key_width = 6,
  )
)
p.save(filename='C:\\Users\\BRB\\geom_raster.png', height=10, width=10, units = 'cm', dpi=400)

结果是：