ThinkPython/structshape.py at v3 · AllenDowney/ThinkPython · GitHub

Name: ThinkPython/structshape.py at v3 · AllenDowney/ThinkPython · GitHub
Rating: 4.5 (6080 reviews)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
"""
This module provides one function, structshape(), which takes
an object of any type and returns a string that summarizes the
"shape" of the data structure; that is, the type, size and
composition.

Copyright 2012 Allen B. Downey
Distributed under the GNU General Public License at gnu.org/licenses/gpl.html.

"""

defstructshape(ds):
"""Returns a string that describes the shape of a data structure.

 ds: any Python object

 Returns: string
 """
typename=type(ds).__name__

# handle sequences
sequence= (list, tuple, set, type(iter('')))
ifisinstance(ds, sequence):
t= []
fori, xinenumerate(ds):
t.append(structshape(x))
rep='%s of %s'% (typename, listrep(t))
returnrep

# handle dictionaries
elifisinstance(ds, dict):
keys=set()
vals=set()
fork, vinds.items():
keys.add(structshape(k))
vals.add(structshape(v))
rep='%s of %d %s->%s'% (typename, len(ds),
setrep(keys), setrep(vals))
returnrep

# handle other types
else:
ifhasattr(ds, '__class__'):
returnds.__class__.__name__
else:
returntypename


deflistrep(t):
"""Returns a string representation of a list of type strings.

 t: list of strings

 Returns: string
 """
current=t[0]
count=0
res= []
forxint:
ifx==current:
count+=1
else:
append(res, current, count)
current=x
count=1
append(res, current, count)
returnsetrep(res)


defsetrep(s):
"""Returns a string representation of a set of type strings.

 s: set of strings

 Returns: string
 """
rep=', '.join(s)
iflen(s) ==1:
returnrep
else:
return'('+rep+')'
return


defappend(res, typestr, count):
"""Adds a new element to a list of type strings.

 Modifies res.

 res: list of type strings
 typestr: the new type string
 count: how many of the new type there are

 Returns: None
 """
ifcount==1:
rep=typestr
else:
rep='%d %s'% (count, typestr)
res.append(rep)


if__name__=='__main__':

t= [1,2,3]
print(structshape(t))

t2= [[1,2], [3,4], [5,6]]
print(structshape(t2))

t3= [1, 2, 3, 4.0, '5', '6', [7], [8], 9]
print(structshape(t3))

classPoint:
"""trivial object type"""

t4= [Point(), Point()]
print(structshape(t4))

s=set('abc')
print(structshape(s))

lt=zip(t, s)
print(structshape(lt))

d=dict(lt)
print(structshape(d))

it=iter('abc')
print(structshape(it))