我有两个列表,如下所示:
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
命令不应该确保转换为数组吗?
最佳答案
这是一种 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()
展平您的阵列,然后减去展平的阵列,然后根据之前的形状重新整形。
关于python - 对 numpy 数组的操作包含不同大小的行,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38894291/