python - 使用 numpy 数组的类设计

Question

我有一个接受三个参数的类：

class Spheroid(object):
def __init__(self, shortaxis, longaxis, height):
    self.shortax = shortaxis
    self.longax  = longaxis
    self.h       = height
    self.alpha   = longaxis/shortaxis

这很好，但有时我喜欢做类似的事情：

heights = np.linspace(0,1,100)

a = Spheroid(1.0,2.0,heights)

a.h

这给了我一个高度数组。

长轴和短轴是我测量的输入参数，但有时我想做类似的事情

class Spheroid(object):
    def __init__(self, alpha, height):
         self.alpha = alpha
         self.h     = height

alphas = np.linspace(0,3,30)

b = Spheroids(alphas, 0.5)

b.alpha 

但我想保留 longax 和 shortax 作为输入参数，所以我不必手动计算 alpha。

关于如何设计这门课来做我想做的事有什么想法吗？