從不等長的數組創建堆疊直方圖

Question

我想創建一個堆疊直方圖。 如果我有一個由三個等長數據集組成的二維數組，這很簡單。 代碼和圖片如下：

import numpy as np
from matplotlib import pyplot as plt

# create 3 data sets with 1,000 samples
mu, sigma = 200, 25
x = mu + sigma*np.random.randn(1000,3)

#Stack the data
plt.figure()
n, bins, patches = plt.hist(x, 30, stacked=True, density = True)
plt.show()

但是，如果我嘗試使用具有不同長度的三個數據集的類似代碼，結果是一個直方圖覆蓋了另一個。 有什么辦法可以用混合長度數據集做堆疊直方圖？

##Continued from above
###Now as three separate arrays
x1 = mu + sigma*np.random.randn(990,1)
x2 = mu + sigma*np.random.randn(980,1)
x3 = mu + sigma*np.random.randn(1000,1)

#Stack the data
plt.figure()
plt.hist(x1, bins, stacked=True, density = True)
plt.hist(x2, bins, stacked=True, density = True)
plt.hist(x3, bins, stacked=True, density = True)
plt.show()

Answer 1

嗯，這很簡單。 我只需要將三個數組放在一個列表中。

##Continued from above
###Now as three separate arrays
x1 = mu + sigma*np.random.randn(990,1)
x2 = mu + sigma*np.random.randn(980,1)
x3 = mu + sigma*np.random.randn(1000,1)

#Stack the data
plt.figure()
plt.hist([x1,x2,x3], bins, stacked=True, density=True)
plt.show()

Answer 2

• 如果pandas是一個選項，則可以將數組加載到數據框中並繪制。
• 使用 pandas 的好處是，數據現在是一種有用的格式，可用於其他分析和其他繪圖。
• 以下代碼將為每個數組創建一個帶有pandas.DataFrame的DataFrames list ，然后將這些數組concat到一個列表理解中。
  • 這是創建長度不相等的數組數據框的正確方法。
    • SO：從條目具有不同長度的字典創建數據幀有更多方法可以從長度不等的數組創建數據幀。
  • 對於等長數組，使用df = pd.DataFrame({'x1': x1, 'x2': x2, 'x3': x3})
• 使用pandas.DataFrame.plot ，它使用matplotlib作為默認繪圖引擎。
  • normed已被matplotlib中的density替換
  • 有關 y 軸值的說明，請參閱matplotlib.pyplot.hist中的density參數。
• 有關其他信息：
  • 繪制直方圖，使條形高度總和為 1（概率）

import pandas as pd
import numpy as np

# create the uneven arrays
mu, sigma = 200, 25
np.random.seed(365)
x1 = mu + sigma*np.random.randn(990, 1)
x2 = mu + sigma*np.random.randn(980, 1)
x3 = mu + sigma*np.random.randn(1000, 1)

# create the dataframe; enumerate is used to make column names
df = pd.concat([pd.DataFrame(a, columns=[f'x{i}']) for i, a in enumerate([x1, x2, x3], 1)], axis=1)

# plot the data
df.plot.hist(stacked=True, bins=30, density=True, figsize=(10, 6), grid=True)