Составьте две гистограммы одновременно с помощью matplotlib - PullRequest
185 голосов
/ 29 июля 2011

Я создал график гистограммы, используя данные из файла, и никаких проблем.Теперь я хотел наложить данные из другого файла в той же гистограмме, поэтому я делаю что-то вроде

n,bins,patchs = ax.hist(mydata1,100)
n,bins,patchs = ax.hist(mydata2,100)

, но проблема в том, что для каждого интервала появляется только столбец с самым высоким значением, а другойскрытый.Интересно, как я мог нарисовать обе гистограммы одновременно разными цветами?

Ответы [ 10 ]

347 голосов
/ 29 июля 2011

Вот вам рабочий пример:

import random
import numpy
from matplotlib import pyplot

x = [random.gauss(3,1) for _ in range(400)]
y = [random.gauss(4,2) for _ in range(400)]

bins = numpy.linspace(-10, 10, 100)

pyplot.hist(x, bins, alpha=0.5, label='x')
pyplot.hist(y, bins, alpha=0.5, label='y')
pyplot.legend(loc='upper right')
pyplot.show()

enter image description here

126 голосов
/ 14 сентября 2016

Принятые ответы дают код для гистограммы с перекрывающимися столбцами, но в случае, если вы хотите, чтобы каждый столбец был рядом (как я сделал), попробуйте вариант ниже:

import numpy as np
import matplotlib.pyplot as plt
plt.style.use('seaborn-deep')

x = np.random.normal(1, 2, 5000)
y = np.random.normal(-1, 3, 2000)
bins = np.linspace(-10, 10, 30)

plt.hist([x, y], bins, label=['x', 'y'])
plt.legend(loc='upper right')
plt.show()

enter image description here

Ссылка: http://matplotlib.org/examples/statistics/histogram_demo_multihist.html

РЕДАКТИРОВАТЬ [2018/03/16]: обновлено, чтобы разрешить построение массивов разных размеров, как предложено @ stochastic_zeitgeist

22 голосов
/ 11 декабря 2017

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

import numpy as np
import matplotlib.pyplot as plt

#makes the data
y1 = np.random.normal(-2, 2, 1000)
y2 = np.random.normal(2, 2, 5000)
colors = ['b','g']

#plots the histogram
fig, ax1 = plt.subplots()
ax1.hist([y1,y2],color=colors)
ax1.set_xlim(-10,10)
ax1.set_ylabel("Count")
plt.tight_layout()
plt.show()

hist_single_ax

В этом случае вы можете нанести два набора данных на разные оси. Для этого вы можете получить данные своей гистограммы с помощью matplotlib, очистить ось, а затем заново нанести ее на две отдельные оси (смещая края корзины таким образом, чтобы они не перекрывались):

#sets up the axis and gets histogram data
fig, ax1 = plt.subplots()
ax2 = ax1.twinx()
ax1.hist([y1, y2], color=colors)
n, bins, patches = ax1.hist([y1,y2])
ax1.cla() #clear the axis

#plots the histogram data
width = (bins[1] - bins[0]) * 0.4
bins_shifted = bins + width
ax1.bar(bins[:-1], n[0], width, align='edge', color=colors[0])
ax2.bar(bins_shifted[:-1], n[1], width, align='edge', color=colors[1])

#finishes the plot
ax1.set_ylabel("Count", color=colors[0])
ax2.set_ylabel("Count", color=colors[1])
ax1.tick_params('y', colors=colors[0])
ax2.tick_params('y', colors=colors[1])
plt.tight_layout()
plt.show()

hist_twin_ax

7 голосов
/ 05 июля 2017

Вот простой метод построения двух гистограмм с их столбцами рядом на одном графике, когда данные имеют разные размеры:

def plotHistogram(p, o):
    """
    p and o are iterables with the values you want to 
    plot the histogram of
    """
    plt.hist([p, o], color=['g','r'], alpha=0.8, bins=50)
    plt.show()
6 голосов
/ 10 декабря 2018

Как завершение ответа Густаво Безерра :

Если вы хотите, чтобы каждая гистограмма была нормализована (normedдля mpl <= 2,1 и <code>density для mpl> = 3,1 ) вы не можете просто использовать normed/density=True, вам нужно вместо этого установить вес для каждого значения:

import numpy as np
import matplotlib.pyplot as plt

x = np.random.normal(1, 2, 5000)
y = np.random.normal(-1, 3, 2000)
x_w = np.empty(x.shape)
x_w.fill(1/x.shape[0])
y_w = np.empty(y.shape)
y_w.fill(1/y.shape[0])
bins = np.linspace(-10, 10, 30)

plt.hist([x, y], bins, weights=[x_w, y_w], label=['x', 'y'])
plt.legend(loc='upper right')
plt.show()

enter image description here

Для сравнения: те же самые векторы x и y с весами по умолчанию и density=True:

enter image description here

3 голосов
/ 31 июля 2018

Вы должны использовать bins из значений, возвращаемых hist:

import numpy as np
import matplotlib.pyplot as plt

foo = np.random.normal(loc=1, size=100) # a normal distribution
bar = np.random.normal(loc=-1, size=10000) # a normal distribution

_, bins, _ = plt.hist(foo, bins=50, range=[-6, 6], normed=True)
_ = plt.hist(bar, bins=bins, alpha=0.5, normed=True)

Two matplotlib histograms with same binning

3 голосов
/ 29 июля 2011

Звучит так, что вам может понадобиться только гистограмма:

Кроме того, вы можете использовать подзаговоры.

2 голосов
/ 16 июня 2017

На случай, если у вас есть панды (import pandas as pd) или вы можете использовать ее:

test = pd.DataFrame([[random.gauss(3,1) for _ in range(400)], 
                     [random.gauss(4,2) for _ in range(400)]])
plt.hist(test.values.T)
plt.show()
0 голосов
/ 18 июня 2019

Вдохновленный ответом Соломона, но придерживайтесь вопроса, который связан с гистограммой, чистое решение:

sns.distplot(bar)
sns.distplot(foo)
plt.show()

Убедитесь, что сначала вычерчивали более высокий, иначе вам нужно было бы установитьplt.ylim (0,0.45), чтобы более высокая гистограмма не была отрублена.

0 голосов
/ 30 апреля 2019

На этот вопрос уже был дан ответ, но я хотел добавить еще один быстрый / простой обходной путь, который может помочь другим посетителям этого вопроса.

import seasborn as sns 
sns.kdeplot(mydata1)
sns.kdeplot(mydata2)

Вот несколько полезных примеров: здесь для kdeСравнение с гистограммой.

...