Многомерная гистограмма в питоне - PullRequest
2 голосов
/ 02 августа 2011

У меня многомерная гистограмма

   H=histogramdd((x,y,z),bins=(nbins,nbins,nbins),range=((0,1),(0,1),(0,1)))

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

Я не знаком с кортежами.Вы можете мне помочь?

1 Ответ

4 голосов
/ 02 августа 2011

используйте, где найти индекс нозеров в H, и используйте индекс, чтобы получить координату:

import numpy as np
x = np.random.random(1000)
y = np.random.random(1000)
z = np.random.random(1000)
nbins = 10
H, [bx, by, bz]=np.histogramdd((x,y,z),bins=(nbins,nbins,nbins),range=((0,1),(0,1),(0,1)))

ix, iy, iz = np.where(H)

for t in zip(bx[ix], by[iy], bz[iz], H[ix,iy,iz]):
    print t
...