numpy.histogram2d

---

---

Примеры

Интерпретация двумерных гистограмм

Отобразить двумерную гистограмму в виде графика можно с помощью библиотеки Matplotlib. В случае одинаковой ширины ячеек можно воспользоваться методом imshow(), например:

>>> import numpy as np
>>> 
>>> x = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         2, 2, 2, 2, 2,
         3, 3, 3,
         4, 4, 4,
         5]
>>> 
>>> y = [1, 1, 1, 1,
         2, 2, 2, 2, 2, 2, 2, 2, 2,
         3, 3, 3, 3, 3, 3,
         4, 4, 4, 4,
         5, 5]
>>> 
>>> 
>>> hist, x_edges, y_edges = np.histogram2d(x, y)
>>> hist
array([[4., 0., 9., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       ...,
       [0., 0., 0., ..., 2., 0., 1.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 1.]])
>>> 
>>> x_edges
array([1. , 1.4, 1.8, ..., 4.2, 4.6, 5. ])
>>> 
>>> y_edges
array([1. , 1.4, 1.8, ..., 4.2, 4.6, 5. ])

Как видим, в результате получилась двумерная гистограмма и два массивами со значениями границ интервалов, причем ширины интервалов по обеим осям одинаковые. Попробуем отобразить данную гистограмму на графике с помощь метода imshow()

>>> import matplotlib.pyplot as plt
>>> 
>>> plt.imshow(hist)
>>> plt.show()

Конечно этот график несет какую-то информацию, но пока ее довольно трудно интерпретировать. Отчасти, дело в автоматическом вычислении границ ячеек. Давайте зададим границы ячеек вручную (границы интервалов являются полуоткрытыми. Подробнее см. в замечанни к функции histogram()):

>>> >>> hist, x_edges, y_edges = np.histogram2d(x, y, bins = 5)
>>> 
>>> plt.imshow(hist)
>>> plt.show()

Теперь данные более сгруппированы, но не мешало бы привести в порядок оси и вывести цветовую карту, что бы можно было сопоставлять цвет с его числовым значением:

>>> plt.imshow(hist.T, origin = 'lower',
... extent = [x_edges[0], x_edges[-1], y_edges[0], y_edges[-1]])
>>> plt.xticks(ticks = x_edges)
>>> plt.xlabel('x')
>>> plt.yticks(ticks = y_edges)
>>> plt.ylabel('y')
>>> plt.colorbar()
>>> plt.show()

Командой hist.T мы транспонировали массив, что бы его строки соответствовали оси y, а столбцы - оси x. с помощью параметра origin = 'lower' мы заставили рисовать гистограмму с левого нижнего угла (это необязательно, но изображение графика в декартовых координатах более привычно).

Теперь займемся интерпретацией получившегося графика двумерной гистограммы, что на самом деле очень просто - давайте добавим на наш график с гистограммой график разброса:

>>> plt.scatter(y, x, color = 'r')
>>> plt.show()

Как видите, каждая пара соответствующих значений из массивов x и y является координатой точки, а функция histogram2d() занимается простым подсчетом количества вхождений точек в заданные прямоугольные области. Так, например, точек с координатами [1, 1] всего 4, точек с координатами [1, 2] больше всего - 9 штук, точек с координатами [2, 3] - 5 штук и т.д.

Границы ячеек

В случае если плоскость необходимо разбить на прямоугольные области разного размера, то отобразить получившуюся гистограмму можно с помощью метода pcolormesh() библиотеки Matplotlib.

Для начала отобразим гистограмму с интервалами ячеек вычисленными автоматически:

>>> rng = np.random.RandomState(777)
>>> 
>>> x = rng.beta(2, 5, size = 1000)
>>> y = rng.normal(1, 4, size = 1000)
>>> 
>>> hist, x_edges, y_edges = np.histogram2d(x, y)
>>> 
>>> hist = hist.T    #  Что бы строки строки соответствовали оси 'y'
>>>                  #  а столбцы оси 'x'
... 
>>> X, Y = np.meshgrid(x_edges, y_edges)
>>> 
>>> plt.pcolormesh(X, Y, hist)
>>> plt.show()

Теперь попробуем задать границы ячеек самостоятельно:

>>> x_edges = np.linspace(x.min(), x.max(), 100)
>>> y_edges = np.linspace(y.min(), y.max(), 20)
>>> 
>>> hist = np.histogram2d(x, y, bins = [x_edges, y_edges])[0]
>>> 
>>> hist = hist.T
>>> 
>>> X, Y = np.meshgrid(x_edges, y_edges)
>>> 
>>> plt.pcolormesh(X, Y, hist)
>>> plt.show()

В некоторых случаях когда точек очень много и на графике разброса происходит их многократное наложение друг на друга, двумерная гистограмма может оказаться абсолютно незаменимой. Например, рассмотрим график разброса на котором 15000 точек:

>>> x = rng.beta(2, 5, size = 15000)
>>> y = rng.normal(1, 4, size = 15000)
>>> 
>>> plt.scatter(x, y, s = 1)
>>> plt.show()

А теперь построим двумерную гистограмму с небольшими ячейками:

>>> x_edges = np.linspace(x.min(), x.max(), 100)
>>> y_edges = np.linspace(y.min(), y.max(), 100)
>>> 
>>> hist = np.histogram2d(x, y, bins = [x_edges, y_edges])[0]
>>> 
>>> hist = hist.T
>>> 
>>> X, Y = np.meshgrid(x_edges, y_edges)
>>> 
>>> plt.pcolormesh(X, Y, hist, cmap = 'nipy_spectral')
>>> plt.colorbar()
>>> plt.show()

Вообще, ширина ячеек может быть задана какой угодно возрастающей функцией:

>>> x = rng.beta(2, 5, size = 10000)
>>> y = rng.normal(1, 4, size = 10000)
>>> 
>>> 
>>> x_edges = np.linspace(x.min()**0.5, x.max()**0.5, 200)**2
>>> y_edges = 0.001*np.linspace(y.min(), y.max(), 75)**3
>>> 
>>> hist = np.histogram2d(x, y, bins = [x_edges, y_edges])[0]
>>> 
>>> hist = hist.T
>>> 
>>> X, Y = np.meshgrid(x_edges, y_edges)
>>> 
>>> plt.pcolormesh(X, Y, hist)
>>> plt.show()

Прочие параметры

Использование остальных параметров данной функции абсолютно аналогично их использованию в функции histogram().