You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
962 B
40 lines
962 B
4 years ago
|
#
|
||
|
|
# 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
|