我遇到了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]]答案 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]]