将常量值添加到2D numpy数组中的列

qv7cva1a  于 12个月前  发布在  其他
关注(0)|答案(1)|浏览(150)

我正在使用ND数组,它包含[x,y,z]空间中多个轮廓的坐标数据。
我尝试根据其他条件向行添加x、y和z偏移量。数组中任何行的偏移量都相同。
下面是一个示例数组

arr= [[[0, 0, 0] 
       [0, 0, 0]]
      [[0, 0, 0]
       [0, 0, 0]]
      [[0, 0, 0]
       [0, 0, 0]]]

字符串
如果我有xOffset = 1,yOffset = 2,zOffset = 0,我如何将它添加到它返回的individal行:

arr= [[[1, 2, 0] 
       [1, 2, 0]]
      [[1, 2, 0]
       [1, 2, 0]]
      [[1, 2, 0]
       [1, 2, 0]]]


我试过用

arr[:][:][0] += xOffset
arr[:][:][1] += yOffset
arr[:][:][2] += zOffset


但它会将偏移量添加到每一列,而不仅仅是子数组中的x,y,z列

iyzzxitl

iyzzxitl1#

你可以尝试创建一个临时的偏移列作为一个1D数组,然后将其添加到正确的列中,你应该能够使用numpy进行列运算(自从我自己尝试以来已经有一段时间了)
这个答案也可能对你有所帮助。
Doing arithmetic with python array columns

相关问题