将带有None值的python列表转换为带有nan值的numpy数组

时间:2013-10-18 18:03:40

标签: python numpy

我正在尝试将包含数值和None值的列表转换为numpy.array,以便None替换为numpy.nan

例如:

my_list = [3,5,6,None,6,None]

# My desired result: 
my_array = numpy.array([3,5,6,np.nan,6,np.nan]) 

天真的方法失败了:

>>> my_list
[3, 5, 6, None, 6, None]
>>> np.array(my_list)
array([3, 5, 6, None, 6, None], dtype=object) # very limited 
>>> _ * 2
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for *: 'NoneType' and 'int'

>>> my_array # normal array can handle these operations
array([  3.,   5.,   6.,  nan,   6.,  nan])
>>> my_array * 2
array([  6.,  10.,  12.,  nan,  12.,  nan])

解决此问题的最佳方法是什么?

2 个答案:

答案 0 :(得分:35)

您只需显式声明数据类型:

>>> my_list = [3, 5, 6, None, 6, None]
>>> np.array(my_list, dtype=np.float)
array([  3.,   5.,   6.,  nan,   6.,  nan])

答案 1 :(得分:4)

怎么样?
my_array = np.array(map(lambda x: numpy.nan if x==None else x, my_list))