简短答案: 请
plt.hist(data, bins=range(50))改用获取左对齐的垃圾箱,
plt.hist(data,bins=np.arange(50)-0.5)获取中心对齐的垃圾箱等。
另外,如果性能很重要,则由于您需要计数唯一的整数,因此
np.bincount我将在后面显示一些更有效的方法()。问题陈述
作为您所看到的独立示例,请考虑以下内容:
import matplotlib.pyplot as pltimport numpy as np# Generate a random array of integers between 0-9# data.min() will be 0 and data.max() will be 9 (not 10)data = np.random.randint(0, 10, 1000)plt.hist(data, bins=10)plt.xticks(range(10))plt.show()
您已经注意到,垃圾箱没有以整数间隔对齐。这基本上是因为您要求 在 0到9 之间的 10个bin ,与要求10个唯一值的bin不太一样。
您想要的箱数与唯一值的数目不完全相同。在这种情况下,您实际应该做的是手动指定纸槽边缘。
为了解释正在发生的事情,让我们跳过
matplotlib.pyplot.hist而仅使用底层
numpy.histogram函数。
例如,假设您有值
[0, 1, 2, 3]。您的第一个本能是:
In [1]: import numpy as npIn [2]: np.histogram([0, 1, 2, 3], bins=4)Out[2]: (array([1, 1, 1, 1]), array([ 0. , 0.75, 1.5 , 2.25, 3. ]))
返回的第一个数组是计数,第二个是bin边(换句话说,条形边在您的绘图中)。
注意,我们得到了期望的计数,但是因为我们要求在数据的最小值和最大值之间有4个bin,所以bin的边缘不在整数值上。
接下来,您可以尝试:
In [3]: np.histogram([0, 1, 2, 3], bins=3)Out[3]: (array([1, 1, 2]), array([ 0., 1., 2., 3.]))
请注意,bin边缘(第二个数组)是您所期望的,但计数不是。这是因为最后一个垃圾箱的行为不同于其他垃圾箱,如文档中所述
numpy.histogram:
Notes-----All but the last (righthand-most) bin is half-open. In other words, if`bins` is:: [1, 2, 3, 4]then the first bin is ``[1, 2)`` (including 1, but excluding 2) and thesecond ``[2, 3)``. The last bin, however, is ``[3, 4]``, which *includes*4.
因此,您实际应该做的是准确指定所需的bin边缘,并在最后一个数据点之外添加一个或将bin边缘移至
0.5间隔。例如:
箱对齐In [4]: np.histogram([0, 1, 2, 3], bins=range(5))Out[4]: (array([1, 1, 1, 1]), array([0, 1, 2, 3, 4]))
现在,将其应用于第一个示例,看看它是什么样的:
import matplotlib.pyplot as pltimport numpy as np# Generate a random array of integers between 0-9# data.min() will be 0 and data.max() will be 9 (not 10)data = np.random.randint(0, 10, 1000)plt.hist(data, bins=range(11)) # <- The only differenceplt.xticks(range(10))plt.show()
好的,太好了!但是,我们现在实际上具有左对齐的垃圾箱。如果我们希望使居中对齐的垃圾箱更好地反映这些是唯一值的事实,该怎么办?
快速的方法是仅移动垃圾箱边缘:
import matplotlib.pyplot as pltimport numpy as np# Generate a random array of integers between 0-9# data.min() will be 0 and data.max() will be 9 (not 10)data = np.random.randint(0, 10, 1000)bins = np.arange(11) - 0.5plt.hist(data, bins)plt.xticks(range(10))plt.xlim([-1, 10])plt.show()
同样,对于右对齐的垃圾箱,只需移动即可
-1。另一种方法
对于唯一整数值的特殊情况,我们可以采用另一种更有效的方法。
如果要处理以0开头的唯一整数,最好使用而
numpy.bincount不是
numpy.hist。
例如:
import matplotlib.pyplot as pltimport numpy as npdata = np.random.randint(0, 10, 1000)counts = np.bincount(data)# Switching to the OO-interface. You can do all of this with "plt" as well.fig, ax = plt.subplots()ax.bar(range(10), counts, width=1, align='center')ax.set(xticks=range(10), xlim=[-1, 10])plt.show()
这种方法有两个很大的优点。一是速度。
numpy.histogram(因此
plt.hist)基本上通过
numpy.digitize然后运行数据
numpy.bincount。因为您要处理唯一的整数值,所以无需执行此
numpy.digitize步骤。
但是,更大的优势是对显示的更多控制。如果您希望使用更薄的矩形,请使用较小的宽度:
import matplotlib.pyplot as pltimport numpy as npdata = np.random.randint(0, 10, 1000)counts = np.bincount(data)# Switching to the OO-interface. You can do all of this with "plt" as well.fig, ax = plt.subplots()ax.bar(range(10), counts, width=0.8, align='center')ax.set(xticks=range(10), xlim=[-1, 10])plt.show()
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)