3

我有一个名为 get_gradient 的简单函数，它接受一个 [[x,y,Vx,Vy]] 的 numpy 数组并返回（应该返回）一个 [[Vx,Vy,Ax,Ay]] 的数组。我想将我的 k[0][0:2] 更新为等于 state[0][2:4] 但是当我将它们打印出来时，k 永远不会改变值。我创建了一个初始数组的副本，希望它能解决它，但它仍然不起作用！任何帮助将非常感激

Y          = numpy.array([[0, 0, 0, 0]]) # [x,y,Vx,Vy]
k          = numpy.array([[0, 0, 0, 0]]) # [x,y,Vx,Vy]

def get_gradient(Y):
    state = Y.copy()
    spring_force   = spring.force(state[0], feet_pos[0])
    brownian_force = mass.brownian(dt)
    drag_force     = mass.drag(state[0][2:4])
    total_force    = spring_force + brownian_force + drag_force     #2d array [Fx,Fy]

    k[0][0:2] = state[0][2:4]            ##### DOESN'T WORK!!!
    k[0][2:4] = total_force/mass.mass    ##### but this works :s

    print k[0][0:2]
    print state[0][2:4]

    return k
4

1 回答 1

4

并不是说 k 没有改变，而是它有一个整数 dtype，所以当你尝试给它分配一个浮点数时，你可能不会得到你所期望的：

>>> k
array([[   0,    0, 4863, 1464]])
>>> k.dtype
dtype('int64')
>>> k[0][0] = 10
>>> k
array([[  10,    0, 4863, 1464]])
>>> k[0][0] = 1e-3
>>> k
array([[   0,    0, 4863, 1464]])

或者

>>> k
array([[  0,    0, 4863, 1464]])
>>> k[0][0] = 10.2
>>> k
array([[  10,    0, 4863, 1464]])

但：

>>> f = k.astype(float)
>>> f
array([[    0.,     0.,  4863.,  1464.]])
>>> f[0][0] = 1e-3
>>> f
array([[  1.00000000e-03,   0.00000000e+00,   4.86300000e+03,
          1.46400000e+03]])

注意.后面的零。

我可能会使用

>>> k = numpy.zeros((1,4))
>>> k
array([[ 0.,  0.,  0.,  0.]])

反而; 它默认为floatdtype。

于 2012-11-17T18:44:58.840 回答