Summary grid
jmp
jeanmichel at sequans.com
Wed Jun 22 12:20:37 EDT 2016
On 06/22/2016 04:46 PM, Jignesh Sutar wrote:
> Say I have list of data as given in the example code below, I want to find
> all the unique categories (alphabetic letters) and unique IDs (numbers) and
> then produce a summary grid as manually entered in the "results". How could
> I code this?
>
> Many thanks in advance,
> Jignesh
>
>
> data= ["A.1", "A.2", "A.3", "B.1", "C.2", "C.3", "D.4", "E.5", "E.6"]
>
> cols=[]
> rows=[]
> for item in data:
> i=item.split(".")
> if i[0] not in cols: cols.append(i[0])
> if i[1] not in rows: rows.append(i[1])
>
> print cols
> print rows
>
> results=
> [["Row/Col", "A", "B", "C", "D", "E"],
> [1, 1, 1, 0, 0, 0],
> [2, 1, 0, 1, 0, 0],
> [3, 1, 0, 1, 0, 0],
> [4, 0, 0, 0, 1, 0],
> [5, 0, 0, 0, 0, 1],
> [6, 0, 0, 0, 0, 1]]
>
Easily done using only builtins and list comprehension:
data= ["A.1", "A.2", "A.3", "B.1", "C.2", "C.3", "D.4", "E.5", "E.6"]
# python magic with the "zip" function
# "set" will take care of removing duplicates
categories, ids = map(set, zip(*[d.split('.') for d in data]))
results = []
for id_ in sorted(map(int,ids)):
results.append([data.count("%s.%d" % (cat, id_)) for cat in
sorted(categories)])
print results
If you don't really understand the zip function:
http://nedbatchelder.com/text/iter.html
The whole presentation is worth reading.
Cheers,
Jm
More information about the Python-list
mailing list