как выполнять операции разброса / сбора в numpy

скажем, у меня есть массивы:

a = array((1,2,3,4,5))
indices = array((1,1,1,1))

и я выполняю операцию:

a[indices] += 1

результат

array([1, 3, 3, 4, 5])

другими словами, дубликаты в индексах игнорируются

, если Я хотел, чтобы дубликаты не игнорировались, что привело бы к следующему:

array([1, 6, 3, 4, 5])

как бы я это сделал?

приведенный выше пример несколько тривиален, я и пытаюсь сделать следующее:

def inflate(self,pressure):
    faceforces = pressure * cross(self.verts[self.faces[:,1]]-self.verts[self.faces[:,0]], self.verts[self.faces[:,2]]-self.verts[self.faces[:,0]])
    self.verts[self.faces[:,0]] += faceforces
    self.verts[self.faces[:,1]] += faceforces
    self.verts[self.faces[:,2]] += faceforces

def constrain_lengths(self):
    vectors = self.verts[self.constraints[:,1]] - self.verts[self.constraints[:,0]]
    lengths = sqrt(sum(square(vectors), axis=1))
    correction = 0.5 * (vectors.T * (1 - (self.restlengths / lengths))).T
    self.verts[self.constraints[:,0]] += correction
    self.verts[self.constraints[:,1]] -= correction

def compute_normals(self):
    facenormals = cross(self.verts[self.faces[:,1]]-self.verts[self.faces[:,0]], self.verts[self.faces[:,2]]-self.verts[self.faces[:,0]])
    self.normals.fill(0)
    self.normals[self.faces[:,0]] += facenormals
    self.normals[self.faces[:,1]] += facenormals
    self.normals[self.faces[:,2]] += facenormals
    lengths = sqrt(sum(square(self.normals), axis=1))
    self.normals = (self.normals.T / lengths).T

Я получил кое-что очень ошибочные результаты в результате игнорирования дубликатов в моих индексированных операциях присваивания.

5
задан damien 29 September 2010 в 03:44
поделиться