Commit c21e72b7 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev

pack tiles inside nested list comprehension

parent b57258c6
#!/usr/bin/env python3
import numpy as np
import itertools
import pack_tile as pile
# 3,3 of (9,9,2) tiles
A = np.zeros((324,242,9,9,5))
d = A.shape
count = 0
for i,j,k,l,m in itertools.product(range(d[0]),range(d[1]),range(d[2]),range(d[3]),range(d[4])):
A[i,j,k,l,m] = count
count += 1
#B = [[[i*5*5+j*5+k for k in range(5)] for j in range(5)] for i in range(5)]
#B = np.array(B)
def pack_tile(tile):
return tile.flatten()
B = np.array([[pack_tile(A[i,j]) for j in range(A.shape[1])] for i in range(A.shape[0])])
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment