將數據着色器圖像添加到 matplotlib 子圖

Question

是否可以將數據着色器圖像添加到一組 matplotlib 子圖中？

作為一個具體的例子，

import numpy as np
import pandas as pd
import matplotlib.pylab as plt

import datashader as ds
import datashader.transfer_functions as tf
from datashader.utils import export_image

from functools import partial

background = "white"
export = partial(export_image, background = background, export_path="export")

N = 10000
df = pd.DataFrame(np.random.random((N, 3)), columns = ['x','y', 'z'])

f, ax = plt.subplots(2, 2)
ax_r = ax.ravel()

ax_r[0].scatter(df['x'], df['y'], df['z'].mean())
ax_r[1].hist(df['x'])
ax_r[2].hist(df['y'])
ax_r[3].plot(df['z'])

cvs = ds.Canvas(plot_width=100, plot_height=100)
agg = cvs.points(df, 'x', 'y', ds.mean('z'))
a = export(tf.shade(agg, cmap=['lightblue', 'darkblue'], how='eq_hist'), 'test')

我有一個ax_r[0]二的 matplotlib 子圖數組，我想用數據着色器圖像a替換上面示例中的 [0,0] 圖ax_r[0] 。 這是可能的，如果是，如何？

謝謝！

Answer 1

更新 [ 2021 年 1 月] 根據下面 James A. Bednar 的評論，Datashader 0.12 現在包括原生 Matplotlib 支持。

截至目前 [ 2017 年 5 月]，將數據着色器圖像添加到 matplotlib 子圖的最佳方法是使用上面鏈接的拉取請求。 它定義了一個DSArtist類。 假設DSArtist類存在，代碼如下：

N = 10000
df = pd.DataFrame(np.random.random((N, 3)), columns = ['x','y', 'z'])

f, ax = plt.subplots(2, 2)
ax_r = ax.ravel()

da = DSArtist(ax_r[0], df, 'x', 'y', ds.mean('z'), norm = mcolors.LogNorm())
ax_r[0].add_artist(da)

ax_r[1].hist(df['x'])
ax_r[2].hist(df['y'])
ax_r[3].plot(df['z'])

plt.tight_layout()
plt.show()