Sophie

Sophie

distrib > Mandriva > 2009.1 > x86_64 > by-pkgid > 72e0913cefe7683427338e51dd70bbde > files > 223

python-cython-0.11.3-1mdv2009.1.x86_64.rpm

#
#  This example demonstrates how to access the internals
#  of a Numeric array object.
#

cdef extern from "Numeric/arrayobject.h":

  struct PyArray_Descr:
    int type_num, elsize
    char type

  ctypedef class Numeric.ArrayType [object PyArrayObject]:
    cdef char *data
    cdef int nd
    cdef int *dimensions, *strides
    cdef object base
    cdef PyArray_Descr *descr
    cdef int flags

def print_2d_array(ArrayType a):
  print "Type:", chr(a.descr.type)
  if chr(a.descr.type) <> "f":
    raise TypeError("Float array required")
  if a.nd <> 2:
    raise ValueError("2 dimensional array required")
  cdef int nrows, ncols
  cdef float *elems, x
  nrows = a.dimensions[0]
  ncols = a.dimensions[1]
  elems = <float *>a.data
  hyphen = "-"
  divider = ("+" + 10 * hyphen) * ncols + "+"
  print divider
  for row in range(nrows):
    for col in range(ncols):
      x = elems[row * ncols + col]
      print "| %8f" % x,
    print "|"
    print divider