数组减法和/或整形

时间:2011-05-13 09:30:41

标签: python arrays numpy operations

我想帮助解决一个问题。在Python中:

a=array([2,2])
b=ones((2,10))

我想知道是否有一个函数允许我减去b-a以得到一个2x10满-1的数组。

我可以用1D数组做一个,我只是想知道是否可以用2D数组。

由于

2 个答案:

答案 0 :(得分:5)

a添加新维度:

b - a[:,None]

其中a[:,None]变为array([[2], [2]]),这是一个2x1数组,您可以从2x10数组中减去并获得一个2x10数组,其中包含-1。

答案 1 :(得分:0)

你想要一个2x10满-1的数组。

你为什么不这样做:

b = np.ones((2, 10)) * -1

array([[-1., -1., -1., -1., -1., -1., -1., -1., -1., -1.],
       [-1., -1., -1., -1., -1., -1., -1., -1., -1., -1.]])