1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
# Copyright (c) 1996, 1997, The Regents of the University of California.
# All rights reserved. See Legal.htm for full text and disclaimer.
# I've felt the need for such a test for a long time;
# this tells you whether an item is a scalar or not.
from types import *
from numpy import ndarray
from scipy import *
def is_scalar (x) :
if type (x) == StringType : return 1
try :
y = len (x)
except (TypeError, AttributeError) :
return 1
return 0
# This routine should be able to tell you the size of any object:
def no_of_dims (x) :
if x == None : return 0
if isinstance(x, ndarray) : return len (x.shape)
if isinstance(x, (list, tuple)) : return 1
# I don't know if there are any other possibilities.
for i in range (10) :
if is_scalar (x) : return i
try :
x = x[0]
except IndexError, TypeError :
return i
return -1
# This routine should define rshape for odd objects:
# It returns a list rather than a tuple; and if it is
# applied to a list, it returns a list each entry of which
# tells the shape of the corresponding list entry, recursively.
def rshape (x) :
ndim = no_of_dims (x)
if ndim == 0 : return 0
s = []
if type (x) == ListType or type (x) == TupleType :
for i in range (len (x)) :
s.append (rshape (x [0]))
x = x [1:]
else : # an array or ???
for i in range ( ndim ) :
s.append (len (x))
x = x[0]
return s
|
