python - 如何在 Matplotlib 中创建 "dot plot"？ (不是散点图)

Question

我想创建我的统计书上所说的“点图”，其中图中的点数等于观测值的数量。这是 mathisfun.com 中的示例:

在示例中，X 轴上 0 值上方有六个点，代表零值的六个观测值。

“点图”似乎可以有多种变化。在查找如何使用 Matplotlib 创建此图时，我只遇到了我所知道的散点图，其中的数据点表示 X 和 Y 值之间的关系。

我尝试使用 Matplotlib 创建的绘图类型是否可行？

Answer 1

假设您有一些数据可以生成如下所示的直方图，

import numpy as np; np.random.seed(13)
import matplotlib.pyplot as plt

data = np.random.randint(0,12,size=72)

plt.hist(data, bins=np.arange(13)-0.5, ec="k")

plt.show()

您可以通过计算直方图并绘制所有可能点的散点图来创建点图，如果点的颜色超过直方图给出的数量，则点的颜色为白色。

import numpy as np; np.random.seed(13)
import matplotlib.pyplot as plt

data = np.random.randint(0,12,size=72)
bins = np.arange(13)-0.5

hist, edges = np.histogram(data, bins=bins)

y = np.arange(1,hist.max()+1)
x = np.arange(12)
X,Y = np.meshgrid(x,y)

plt.scatter(X,Y, c=Y<=hist, cmap="Greys")

plt.show()

或者，您可以将不需要的点设置为nan，

Y = Y.astype(np.float)
Y[Y>hist] = np.nan

plt.scatter(X,Y)