Накресліть дві гістограми на одному діаграмі з matplotlib


233

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

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

але проблема полягає в тому, що для кожного інтервалу відображається лише смужка з найвищим значенням, а інша прихована. Цікаво, як я міг будувати обидві гістограми одночасно різними кольорами.

Відповіді:


418

Ось у вас є робочий приклад:

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()

введіть тут опис зображення


1
Чи не було б гарною ідеєю встановити pyplot.hold(True)перед складанням планів, про всяк випадок?
JAB

2
Не впевнений, чи встановлено hold (True) у моїх параметрах matplotlib config або pyplot поводиться так за замовчуванням, але для мене код працює як є. Код витягується з більшого додатку, який поки не створює жодних проблем. У всякому разі, гарне запитання, яке я вже робив собі при написанні коду
Joaquin

@joaquin: як я можу вказати x на синій, а y на червоний?
amc

7
Коли я відтворив сюжет із кольоровим кольором брусків, Noneза замовчуванням. Якщо ви бажаєте того ж дизайну, який показаний на графіку, ви можете встановити edgecolorпараметр як в, наприклад, k(чорний). Процедура схожа на легенду.
Тож S

2
Ще простіше: pyplot.hist([x, y], bins, alpha=0.5, label=['x', 'y']).
Августин

174

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

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()

введіть тут опис зображення

Довідка: http://matplotlib.org/examples/statistics/histogram_demo_multihist.html

EDIT [2018/03/16]: оновлено, щоб дозволити побудову масивів різних розмірів, як запропонував @stochastic_zeitgeist


@GustavoBezerra, як використовувати plt.histдля створення одного файлу PDF для кожної гістограми? Я завантажив свої дані за допомогою, pandas.read_csvі файл має 36 стовпців і 100 рядків. Тому я хотів би 100 pdf-файлів.
Сигур

2
@ Сигур Це зовсім поза темою. Будь ласка, Google або задайте нове запитання. Здається, це пов’язано: stackoverflow.com/questions/11328958/…
Густаво Безерра

1
@stochastic_zeitgeist Я згоден з @pasbi. Я використовував ваш коментар із фреймом даних панди, тому що мені потрібні були різні ваги через нан. з, x=np.array(df.a)і в y=np.array(df.b.dropna())основному це закінчилосяplt.hist([x, y], weights=[np.ones_like(x)/len(x), np.ones_like(y)/len(y)])
grinsbaeckchen

1
У випадку, якщо розміри вибірки різко відрізняються, ви можете побудувати схему, використовуючи дві осі, щоб краще порівняти розподіли. Дивіться нижче .
Андрій

1
@ AgapeGal'lo Будь ласка, зверніться до відповіді Ендрю.
Густаво Безерра

30

У випадку, якщо ви маєте різні розміри вибірки, може бути важко порівняти розподіли з однією віссю у. Наприклад:

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


1
Це хороша коротка відповідь, за винятком того, що ви також повинні додати, як центрувати смуги на кожній мітці
Odisseo

12

Як доповнення до відповіді Густаво Безерри :

Якщо ви хочете, щоб кожна гістограма була нормалізована ( normedдля mpl <= 2.1 і 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()

введіть тут опис зображення

Для порівняння - точно такі ж xі yвектори з типовими вагами і density=True:

введіть тут опис зображення


9

Вам слід використовувати 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)

Дві гістограми matplotlib з однаковим бінінгуванням


7

Ось простий метод побудувати дві гістограми з їхніми брусками на одному графіку, коли дані мають різні розміри:

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()

3

Це здається, що ви хочете просто графік:

Як варіант, ви можете використовувати підзаготівлі.


Різниця полягає в тому, що з історією ви отримуєте графік частоти. можливо, ви повинні показати, як це зробити. частота з пандами + смуговий графік = hist ()
VP.

2

Про всяк випадок, якщо у вас є панди ( 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()

Я вважаю, що використання панд не вийде, якщо гістограми для порівняння мають різні розміри вибірки. Це також часто контекст, в якому використовуються нормовані гістограми.
Соломон Вімал

2

Є одне застереження, коли ви хочете побудувати гістограму з 2-денного масивного нумеру. Потрібно поміняти місцями 2 осі.

import numpy as np
import matplotlib.pyplot as plt

data = np.random.normal(size=(2, 300))
# swapped_data.shape == (300, 2)
swapped_data = np.swapaxes(x, axis1=0, axis2=1)
plt.hist(swapped_data, bins=30, label=['x', 'y'])
plt.legend()
plt.show()

введіть тут опис зображення


0

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

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

Деякі корисні приклади тут для KDE проти порівняння гістограм.


0

Натхненний відповіддю Соломона, але, щоб дотримуватися питання, пов'язаного з гістограмою, чистим рішенням є:

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

Переконайтеся, що спочатку побудуйте вище, інакше вам потрібно встановити plt.ylim (0,0.45), щоб гістограма вище не була відсічена.


0

Також варіант, який досить схожий на жокейн відповідь:

import random
from matplotlib import pyplot

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

#plot both histograms(range from -10 to 10), bins set to 100
pyplot.hist([x,y], bins= 100, range=[-10,10], alpha=0.5, label=['x', 'y'])
#plot legend
pyplot.legend(loc='upper right')
#show it
pyplot.show()

Дає такий вихід:

введіть тут опис зображення

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.