python - 对 numpy 数组的操作包含不同大小的行

Question

我有两个列表，如下所示:

a= [[1,2,3,4], [2,3,4,5],[3,4,5,6,7]], b= [[5,6,7,8], [9,1,2,3], [4,5,6,7,8]]

对于这样的输出，我想逐个元素地减去:

a-b= [[-4,-4,-4,-4],[7,2,2,2],[-1,-1,-1,-1,-1]]

为了做到这一点，我将每个 a 和 b 转换为数组并减去它们，我使用:

np.array(a)-np.array(b)

输出只是给我错误:

Unsupported Operand type for-: 'list' and 'list'

我做错了什么？ np.array 命令不应该确保转换为数组吗？

Answer 1

这是一种 Numpythonic 方式:

>>> y = map(len, a)  
>>> a = np.hstack(np.array(a))
>>> b = np.hstack(np.array(b))
>>> np.split(a-b, np.cumsum(y))
[array([-4, -4, -4, -4]), array([-7,  2,  2,  2]), array([-1, -1, -1, -1, -1]), array([], dtype=float64)]
>>> 

由于您不能减去具有不同形状的数组，您可以使用 np.hstack() 展平您的阵列，然后减去展平的阵列，然后根据之前的形状重新整形。