具有2D功能的Python十六进制图

发布于05月06日

我试图用pyplot.hexbin在六边形网格上显示二维函数，但它只生成一条线并忽略了函数的其余部分.我该如何解决这个问题？

def func(x,y):
    return x*2+y*2

x = np.linspace((-0.5)/4*5, (0.5)/4*5, int(2e3))   
y = np.linspace(-0.5, 0.5, int(2e3))    

plt.hexbin(x,
           y,
           func(x,y), 
           gridsize=(8,4),
           cmap='gnuplot')
plt.show()

enter image description here

推荐答案

func(x，y)函数 for each (x，y)对返回一个纯量值，但hexbin期望x和y是代表坐标的相同长度的array.输入另一个带有纯量值的数组Z，这些值将是每个十六进制的 colored颜色 . try :

import numpy as np
import matplotlib.pyplot as plt

def func(x,y):
    return x*2+y*2

x = np.linspace((-0.5) / 4 * 5, (0.5) / 4 * 5, int(2e3))
y = np.linspace(-0.5, 0.5, int(2e3))
X, Y = np.meshgrid(x, y)

# Evaluate the function for each combination of x and y
Z = func(X, Y)

plt.hexbin(X.flatten(),    # Flatten X and Y to 1D arrays
           Y.flatten(),
           C=Z.flatten(),  # Flatten Z to 1D array for color values
           gridsize=(8, 4),
           cmap='gnuplot')

plt.show()

您还可以在演出前添加彩色条，例如: