从二维矩阵创建密度图

Question

我有一个二维值矩阵，我想将其绘制为二维直方图。

一个简化的例子：我有一个初始速度的一维数组，比如 vi = [1, 2, 3]，对于 vi 中的每个值，我有一行相应的最终速度存储在二维数组中，vf = [ [0.7, 1.1, 1.5], [1.8, 2.1, 2.4], [2.7, 2.9, 3.1] ]。我希望能够制作点 (vi, vf) 的二维直方图，即坐标 [1, 0.7], [1, 1.1], [1, 1.5], [2, 1.8], [2, 2.1], [2, 2.4], and [3,2.7], [3, 2.9], [3, 3.1].

有没有办法创建这样的配对？

The answer to this question 建议使用 imshow 或 matshow，但会根据分配给每个元素的值对 bin 进行着色。我需要的是一个绘图例程，它采用二维矩阵，将其分成一个网格，并通过每个箱子中的 count 为每个箱子着色。

感谢任何帮助！

Answer 1

您似乎有一个 2D space，其中 x 值来自 vi，y 值来自 vf。重复 vi n 次（n 是 vf 的行长）使得 x 和 y 数组具有相同数量的元素，对应于所需的元组。

在代码中：

import numpy as np
import seaborn as sns
from matplotlib import pyplot as plt

vi = np.array([1, 2, 3])
vf = np.array([[0.7, 1.1, 1.5], [1.8, 2.1, 2.4], [2.7, 2.9, 3.1]])

x = np.repeat(vi, vf.shape[1]) # repeat the x-values by the row-length of `vf`
y = vf.ravel() # convert to a 1D array
sns.histplot(x=x, y=y)
plt.show()

这么少的数据，情节看起来很无趣。您必须使用真实数据进行测试，以确定它是否符合您的预期。

print([*zip(x,y)]) 打印 (x,y) 作为元组，即

[(1, 0.7), (1, 1.1), (1, 1.5), (2, 1.8), (2, 2.1), (2, 2.4), (3, 2.7), (3, 2.9), (3, 3.1)]

从二维矩阵创建密度图

Create density plot from 2d matrix

python

matplotlib

matrix

imshow