0

我需要在 numpy 中计算一些方向数组。我将 360 度分为 16 组，每组覆盖 22.5 度。我想要组中间的 0 度，即获取 -11.25 度和 11.25 度之间的方向。但问题是我怎样才能得到 168.75 度和 -168.75 度之间的组？

a[numpy.where(a<0)] = a[numpy.where(a<0)]+360
for m in range (0,3600,225):
        b = (a*10 > m)-(a*10 >= m+225).astype(float)    
        c = numpy.apply_over_axes(numpy.sum,b,0)
4

1 回答 1

1

如果你想把数据分成16组，中间有0度，你为什么要写for m in range (0,3600,225)？

>>> [x/10. for x in range(0,3600,225)]
[0.0, 22.5, 45.0, 67.5, 90.0, 112.5, 135.0, 157.5, 180.0, 202.5, 225.0, 247.5,
 270.0, 292.5, 315.0, 337.5]
## this sectors are not the ones you want!

我会说你应该从for m in range (-1125,36000,2250)（注意，现在我使用的是 100 因子而不是 10）开始，这会给你你想要的组......

wind_sectors = [x/100.0 for x in range(-1125,36000,2250)]
for m in wind_sectors:
    #DO THINGS

我不得不说我不太了解你的脚本和它的目标......为了处理圆形度数，我建议如下：

  • 一种条件，您将有问题的数据放置在其中，即您必须处理零附近的转换；
  • 放置所有其他数据的条件。

例如，在这种情况下，我打印数组中属于每个扇区的所有元素：

import numpy

def wind_sectors(a_array, nsect = 16):
    step = 360./nsect
    init = step/2
    sectores = [x/100.0 for x in range(int(init*100),36000,int(step*100))]

    a_array[a_array<0] = a_arraya_array[a_array<0]+360

    for i, m in enumerate(sectores):
        print 'Sector'+str(i)+'(max_threshold = '+str(m)+')'
        if i == 0:
            for b in a_array:
                if b <= m or b > sectores[-1]:
                    print b

        else:
            for b in a_array:
                if b <= m and b > sectores[i-1]:
                    print b
    return "it works!"

# TESTING IF THE FUNCTION IS WORKING:
a = numpy.array([2,67,89,3,245,359,46,342])

print wind_sectors(a, 16)

# WITH NDARRAYS:

b = numpy.array([[250,31,27,306], [142,54,260,179], [86,93,109,311]])

print wind_sectors(b.flat[:], 16)

关于 flat 和 reshape 功能：

>>> a = numpy.array([[0,1,2,3], [4,5,6,7], [8,9,10,11]])
>>> original = a.shape
>>> b = a.flat[:]
>>> c = b.reshape(original)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> b
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
>>> c
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
于 2012-05-23T11:20:32.580 回答