在热图网格中绘制具有4个参数的函数

Question

我在python 2中使用plotnine（但是很乐意使用matplotlib或任何其他绘图包来解决）。我有一个函数（下面稍微简化）有4个参数。我想绘制一个热图网格，用＃＆＃34;超轴＆＃34;改变两个参数，每个热图改变另外两个参数。有点像这样：

我想出了这段代码：

from plotnine import ggplot, aes, facet_grid, geom_tile
import pandas as pd
import itertools

def fun((i, j, n, p)):
    if n > j:
        return 1 if (p**(3*n)) > ((p+i)**(3*(n-j))) else 0
    return -1

ilist, jlist, nlist, plist = range(1,10), range(1,9), range(8,10), range(4,6)
rows = itertools.product(ilist, jlist, nlist, plist)

df = pd.DataFrame(list(rows))
df.columns = ['i','j','n','p']
df['fun'] = df.apply(fun, axis=1)

(ggplot(df, aes('factor(i)', 'factor(j)')) +
 geom_tile(aes(fill='fun')) +
facet_grid('n ~ p', labeller='label_both'))

这会产生以下结果：

这个（或多或少）具有我想要的形式，但是热图似乎是错误的。 （编辑：我发现错误出现在fun的定义中，而不是在绘图中 - 请参阅我的回答。）

Answer 1

好的......原来这不是绘图的问题，而是功能。我不完全明白出了什么问题，但我认为有些计算遇到了“巨大的整数”问题。当我运行它以在日志空间中进行比较时，它开始工作。

我会把它留在这里，因为它可能会帮助其他人试图制作类似的热图网格，也许有人会以更好的方式回答这个问题。

以下是我带注释的代码，用于从包含['i', 'j', 'n', 'p']列的数据框中创建热图：

# The first two lines are defining the subplots, i.e. the heatmaps
# themselves. I am passing the dataframe, and specifying the names of the
# columns to use as axes. I say 'factor(i)' etc to treat the i column as
# discrete, not continuous (in my case it's integers).
gg = (ggplot(df, aes('factor(i)', 'factor(j)')) +
    geom_tile(aes(fill='fun')) +
# The last bit is to call facet_grid which applies the above code
# in a grid. The parameter 'n ~ p' specifies that I want the grid to be
# over the columns 'n' and 'p' from the dataframe. The labeller
# parameter is what makes the labels at the edges (see top and right in
# image below) show both the column name and the value).
    facet_grid('n ~ p', labeller='label_both'))

以下是具有固定结果的完整更正代码：

from plotnine import ggplot, aes, facet_grid, geom_tile
import pandas as pd
import numpy as np
import itertools

def fun((i, j, n, p)):
    if n > j:
        return 1 if np.log10(p) * (3*n) > np.log10(p+i) * (3*(n-j)) else 0

    return -1

ilist, jlist, nlist, plist = range(1,10), range(1,9), range(8,10), range(4,6)
rows = itertools.product(ilist, jlist, nlist, plist)
df = pd.DataFrame(list(rows))
df.columns = ['i','j','n','p']
df['fun'] = df.apply(fun, axis=1)

gg = (ggplot(df, aes('factor(i)', 'factor(j)')) +
    geom_tile(aes(fill='fun')) +
    facet_grid('n ~ p', labeller='label_both'))

gg.draw()