Не построение "нуля" в matplotlib или изменение нуля на None [Python]

у меня есть код ниже, и я хотел бы преобразовать все нули в данные в None(поскольку я не хочу строить данные здесь, в matplotlib). Однако, код notworking и 0. все еще печатается

sd_rel_track_sum=np.sum(sd_rel_track, axis=1)
for i in sd_rel_track_sum:
   print i
   if i==0:
       i=None

return sd_rel_track_sum

может ли кто-нибудь придумать решение этого. Или просто ответ на то, как я могу перенести все 0 в None. Или просто не построить нулевые значения в Matplotlib.

2 ответов


values = [3, 5, 0, 3, 5, 1, 4, 0, 9]

def zero_to_nan(values):
    """Replace every 0 with 'nan' and return a copy."""
    return [float('nan') if x==0 else x for x in values]

print(zero_to_nan(values))

дает вам:

[3, 5, nan, 3, 5, 1, 4, nan, 9]

Matplotlib не будет строитьnan (не число) значений.


почему бы не использовать numpy для этого?

>>> values = np.array([3, 5, 0, 3, 5, 1, 4, 0, 9], dtype=np.double)
>>> values[ values==0 ] = np.nan
>>> values
array([  3.,   5.,  nan,   3.,   5.,   1.,   4.,  nan,   9.])

следует отметить, что значения не могут быть массив целых чисел тип.