0

我不知道如何使用 CArray 特征。为什么这门课

from traits.api import HasTraits, CArray, Float,Int
import numpy as np

class Coordinate3D(HasTraits):
   coordinate = CArray(Float(), shape=(1,3) )

   def _coordinate_default(self):
     return np.array([1.,2.,3.])

显然不使用我的 _name_default() 方法?

In [152]: c=Coordinate3D()
In [153]: c.coordinate
Out[153]: np.array([[ 0.,  0.,  0.]])

我本来期望 np.array([1,2,3]) !_name_default() 似乎适用于 Int

class A(HasTraits):
    a=Int
    def _a_default(self):
       return 2

In [163]: a=A()    
In [164]: a.a
Out[164]: 2

那么我在这里做错了什么?另外,我不能赋值:

In [181]: c.coordinate=[1,2,3]
TraitError: The 'coordinate' trait of a Coordinate3D instance must be an array of      
float64 values with shape (1, 3), but a value of array([ 1.,  2.,  3.]) <type
'numpy.ndarray'> was specified.

与相同的错误消息

In [182]: c.coordinate=np.array([1,2,3])
4

3 回答 3

2

一维数组和其中一个维度的大小为 1 的二维数组之间存在差异。您正在尝试将一维数组设置为CArray期望二维的特征。例如,您的默认方法应该是:

def _coordinate_default(self):
    return np.array([[1., 2., 3.]])

(注意额外的方括号)。您设置的阵列是形状(3,),而不是所需的(1, 3)

同样,它不会将平面列表强制转换为二维数组。尝试分配一个嵌套列表,例如

c.coordinate=[[1, 2, 3]]

反而。

(或者,如果你真的想要一维数组,你应该shape=(3,)在你的特征分配中使用,其他部分应该可以正常工作。)

于 2014-01-08T14:09:12.913 回答
0

骗我。在从 Eclipse 复制粘贴到 iPython 时,我没有使用神奇的 %paste 函数并在那里弄乱了类定义。另一个实际错误是 CArray 的形状,它必须是 (3,)。

这段代码

class Coordinate3D(HasTraits):  

   coordinate = CArray(Float(),shape=(3,))

   def __init__(self,iv=None):
     super(Coordinate3D,self).__init__()
     if iv:
       self.coordinate=iv

   def _coordinate_default(self):
     return array([1,2,3])

   def __getitem__(self,index):
     return self.coordinate[index]

像预期的那样工作:

In [3]: c=Coordinate3D()
In [6]: c.coordinate
Out[6]: array([ 1.,  2.,  3.])

In [7]: c=Coordinate3D([1,2,5])
In [8]: c.coordinate
Out[8]: array([ 1.,  2.,  5.])

In [11]: c[0]
Out[11]: 1.0
于 2014-01-08T14:20:52.277 回答
0

作为对先前答案的扩展,我进一步进行了实验:

import types
RealNumberType    = (types.IntType, types.LongType, types.FloatType)

class ScaleFactor3D(Coordinate3D):
  '''Demonstrate subclassing a HasTraits class
     and overriding __init__ and a _default method''' 
  def _coordinate_default(self):
    return array([1,1,1])

  def __init__(self,iv=None):
    if isinstance(iv,RealNumberType):
      iv=[iv,iv,iv]
    super(ScaleFactor3D,self).__init__(iv)  

这也很好用:

In [35]: s=ScaleFactor3D()

In [36]: s.coordinate
Out[36]: array([ 1.,  1.,  1.])

In [37]: s=ScaleFactor3D(3)

In [38]: s.coordinate
Out[38]: array([ 3.,  3.,  3.])

我想我会把它放在这里,因为我在网上找不到太多关于 CArray 的有用信息。

于 2014-01-08T14:36:02.397 回答