ヒストグラムのデータを抽出したい

Question

前提・実現したいこと

サイトのプログラムを使って作成したヒストグラムの各データを取得するプログラムを作ろうと思っています。

発生している問題・エラーメッセージ

ヒストグラムを作っているプログラムからデータを持ってくる方法がわからなかったため、r,g,b値を08,916...と8個ずつ、合計することでヒストグラムと同じデータを作ろうとしましたが、ヒストグラムとデータの個数が合いませんでした。例えばヒストグラムだとGreenの32付近は15000近くデータがあるのに対し、作成したプログラムでは、1000個ほどしか得られませんでした。

該当のソースコード

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

im = Image.open('lena_square.png')

plt.figure()
r = np.array(im)[:, :, 0].flatten()
g = np.array(im)[:, :, 1].flatten()
b = np.array(im)[:, :, 2].flatten()

bins_range = range(0, 257, 8)
xtics_range = range(0, 257, 32)

plt.hist((r, g, b), bins=bins_range,
         color=['r', 'g', 'b'], label=['Red', 'Green', 'Blue'])
plt.legend(loc=2)

plt.grid(True)

[xmin, xmax, ymin, ymax] = plt.axis()
plt.axis([0, 256, 0, ymax])
plt.xticks(xtics_range)
plt.savefig("matplotlib_histogram_single.png")

r_sum = 0
g_sum = 0
b_sum = 0
r_count = 0
g_count = 0
b_count = 0

for i in range(0,257):
	r_sum = r_sum + r[i]
	g_sum = g_sum + g[i]
	b_sum = b_sum + b[i]
	if i%8==0:
		r_count = np.append(r_count,r_sum)
		g_count = np.append(g_count,g_sum)
		b_count = np.append(b_count,b_sum)
		r_sum = 0
		g_sum = 0
		b_sum = 0

print(r_count)

ヒストグラムから直接データを抽出する方法、もしくは作成したプログラムの修正点を教えていただきたいです。

Answer 1

ret = plt.hist()とすると、retに度数と階級値を取得することができます。先の場合だと、ret[0]が度数のarrayになり、ret[1]が階級値のarrayになります。

なお、ヒストグラムの度数と階級値を計算するだけであれば、np.histogram()も使えます。

Comments

[uni_tt]

解決しましたありがとうございます。