matplotlib中的离散密度图

Question

我有一个2D numpy数组，我想用它创建一个离散的密度图。 在图中的每个点(i,j)应该放置一个点，其中颜色应该对应于2D阵列的第(i,j)个元素的值。 我不想使用imshow因为我不想要任何插值，我也想控制要放置的点的大小。

Answer 1

你试过imshow with interpolation='nearest'吗？ 这接近你想要的吗？

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(100).reshape(10, 10)
fig, ax = plt.subplots()
ax.imshow(data, interpolation='nearest')
plt.show()

Answer 2

您可以随时为每个点明确地执行此操作，但这将很慢。 我认为最好一行一行（保持说明固定）并使用scatter功能。

import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D

n = 100
x = plt.linspace(0,5, n)
y = plt.linspace(0,5, n)

ax = plt.subplot(111)
for i in range(n):
    y_fixed = y[i] * plt.ones(n)
    z = [(abs(plt.cos(x[i])), 0.0, 0.5) for i in range(n)]
    ax.scatter(x, y_fixed, c=z)

plt.show()

使用s参数在该庄园中也可以调整大小。

如果没有任何关于指定颜色的数据，我使用的是RGB值。 您可能需要标准化，但是c=将采取任何相当多的东西并将其转换为颜色，但这可能与您无关。

有关散射的更多信息，请参阅此处的演示