Функция игнорировать значения нанов и делать numpy.polyval - PullRequest
Новая
       10

Функция игнорировать значения нанов и делать numpy.polyval

0 голосов
/ 25 мая 2018

Я пытаюсь закодировать функцию, в которую вы передаете ей массив mxn, его цель - выполнить итерацию по массиву и выполнить numpy.polyval для каждого значения.Массивы будут содержать некоторые значения nan, поэтому я хочу, чтобы функция возвращала значение nan в том же порядке, в каком оно было в исходном массиве.Вот моя функция до сих пор: 

def polyval_NaN(p,data):
    m,n = data.shape
    data2 = data
    for i in np.arange(1,m):
        for j in np.arange(1,n):
            if np.isnan(data2[i,j]) == False:
                data2[i,j] = np.polyval(p,data[i,j])
    return data2

Спасибо за вашу помощь

1 Ответ

0 голосов
/ 25 мая 2018

Используйте логическое индексирование следующим образом: 

>>> import numpy as np
>>> 
>>> p = np.array((4,5,3))
>>> a = np.arange(12.).reshape(3,4)
>>> a.ravel()[[2,5,7]] = np.nan
>>> a
array([[ 0.,  1., nan,  3.],
       [ 4., nan,  6., nan],
       [ 8.,  9., 10., 11.]])
>>> 
>>> result = np.full_like(a, np.nan)
>>> good = ~np.isnan(a)
>>> result[good] = np.polyval(p, a[good])
>>> 
>>> result
array([[  3.,  12.,  nan,  54.],
       [ 87.,  nan, 177.,  nan],
       [299., 372., 453., 542.]])
...