如何使用python-numpy-broadcasting

时间:2016-07-07 05:34:26

标签: python numpy numpy-broadcasting

我遇到了python广播的问题

有两个numpy列表

x = np.array([[1,2,3],[4,5,6]])

y = np.array([0,1,1])

我想计算

x : [[1,2,3],
     [4,5,6]]

x : [[1-y[0], 2-y[1], 3-y[2]],
     [4-y[0], 5-y[1], 6-y[2]]]

x : [[0,-3,-3],
     [3,0,0]] 

一次。

请让我知道如何做到这一点。

对不起......我有一个错误......

我的意思是     x:[[1,1,2],          [4,4,5]]

1 个答案:

答案 0 :(得分:2)

我不知道你是怎么得到的:

x : [[0,-3,-3],
     [3,0,0]] 

但要做到你要求的最简单方法就是:

import numpy as np
X = ([1,2,3],[4,5,6])
y = np.array([0,1,1])
print X-y

输出:

  

[[1 1 2]
   [4 4 5]]