zéro les valeurs d'un tableau à être converti en valeurs nan

J'ai un tableau de 1200*1200. Certains de ses valeurs sont égales à zéro. Je veux convertir les valeurs nulles de numpy.nan valeurs. C'est ma solution:

import numpy
for i in range(1200):
    for j in range(1200):
        if data_a[i, j] == 0:
            data_a[i, j] = numpy.nan

Mais j'ai eu cette erreur: data_a[i,j] = numpy.nan
ValueError: cannot convert float NaN to integer
Je ne comprends pas l'erreur. Des alternatives ou des solutions?

OriginalL'auteur f.ashouri | 2012-11-09

  1. 6

    Que le message d'erreur est parce que votre tableau est pour stocker des entiers:

    >>> import numpy as np
>>> a = np.arange(3)
>>> a
array([0, 1, 2])
>>> a.dtype
dtype('int32')
>>> a[0] = np.nan
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: cannot convert float NaN to integer

    Si votre tableau est de flotteurs, cependant, il va fonctionner. Vous pouvez le faire sans boucles, trop:

    >>> a = np.arange(3.0)
>>> a
array([ 0.,  1.,  2.])
>>> a[a==0]
array([ 0.])
>>> a[a==0] = np.nan
>>> a
array([ nan,   1.,   2.])

    Si vous voulez convertir votre tableau de float tableau, vous pouvez utiliser astype:

    >>> a = a.astype(float)
>>> a
array([ 0.,  1.,  2.])
>>> a.dtype
dtype('float64')

    OriginalL'auteur DSM