您参差不齐的数组 - 列表数组:
In [31]: arr = np.array([[1,2,3],[1.1,6],[1]])
<ipython-input-31-4887f672b831>:1: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.
arr = np.array([[1,2,3],[1.1,6],[1]])
In [32]: arr
Out[32]: array([list([1, 2, 3]), list([1.1, 6]), list([1])], dtype=object)
对象 dtype 数组的数学运算是逐个元素执行的(以列表理解速度)。它如何工作取决于相应的方法如何工作元素:
In [33]: arr *2
Out[33]:
array([list([1, 2, 3, 1, 2, 3]), list([1.1, 6, 1.1, 6]), list([1, 1])],
dtype=object)
list*2
是复制操作,而不是乘法。
In [34]: arr / 2
Traceback (most recent call last):
File "<ipython-input-34-855a165721c4>", line 1, in <module>
arr / 2
没有为列表定义除法。
TypeError: unsupported operand type(s) for /: 'list' and 'int'
我们可以对每个元素应用一个简单的函数frompyfunc
:
In [38]: np.frompyfunc(lambda x: np.array(x)/2, 1,1)(arr)
Out[38]:
array([array([0.5, 1. , 1.5]), array([0.55, 3. ]), array([0.5])],
dtype=object)
此函数负责转换为数组以及除法。
等效的列表理解(同样快):
In [40]: [np.array(x)/2 for x in arr]
Out[40]: [array([0.5, 1. , 1.5]), array([0.55, 3. ]), array([0.5])]
或纯列表版本(可能更快)
In [41]: [[y/2 for y in x] for x in arr.tolist()]
Out[41]: [[0.5, 1.0, 1.5], [0.55, 3.0], [0.5]]