Как включить нан весов в NumPy Bincounts - PullRequest
0 голосов
/ 16 июня 2019

Функция bincount Numpy дает значение nan, если одно значение в весах равно nan.Я только хочу, чтобы он возвращал значение nan, если все веса равны nan для этого бина.

Использование модифицированной версии этого поста ( Самый эффективный способ вычисления радиального профиля ) Я пытаюсьсоздать среднее радиальное значение, но при этом хотеть включить области с разреженными, но все же представленными данными в среднем, но показывать пропущенные значения, когда данных вообще нет

def radial_profile(data, center):
    y, x = np.indices((data.shape))
    r = np.sqrt((x - center[0])**2 + (y - center[1])**2)
    r = r.astype(np.int)
    DR=data.ravel()
    RR=r.ravel()
    keep = ~np.isnan(DR)
    tbin = np.bincount(RR[keep],DR[keep])
    rbin = np.bincount(RR[keep])
    tr=tbin/rbin
    return tr

Это позволяет мне усреднять данные радиально, несмотря напропущены некоторые значения, но не указывается значение nan при пустом радиусе данных.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...