Creating list of lists containing objects in python

Question

I am trying to put a custom object into a 2d list and I'm getting very confused. Here's what I have:

Here's the cell class:

class Cell(object):
    def __init__(self, what, cost):
        self.what = what
        self.parentx = -1
        self.parenty = -1
        self.f = 0
        self.g = 0
        self.h = 0
        self.cost = cost

and my insertion object (where I take a grid/ maze and put the data into objects):

def insertion (r, c, grid):
    cellGrid = []
    for x in range(0, r):
        for y in range(0, c):
            if (grid[x][y] == '%'):
                what = 0
                cost = 100000000
            elif(grid[x][y] == '-'):
                what = 1
                cost = 1
            elif (grid[x][y] == '.'):
                what = 2
                cost = 0
            else:
                what = 3
            cellGrid.append(Cell(what, cost))
    return cellGrid

This does work, but what I really want is cellGrid to be a 2d list, just like grid, except that it will contain cell objects intstead of a string. I would like to be able to acceess it as cellGrid[0][0].what for example. How should I accomplish this?

Don mix tabs and spaces - in particular, never do that in Python code. You might use tabs, but it is preferable to use just spaces - MUCH preferable. — jsbueno
– jsbueno, Commented May 14, 2014 at 3:52

user764357 · Accepted Answer · 2014-05-14 03:57:25Z

2

If you want to do that you need to create a 'row' in the first loop to add into the main grid array. You then append cells to this row, and append the whole row to the grid.

Like so:

def insertion (r, c, grid):
    cellGrid = []
    for x in range(0, r):
        row = []
        for y in range(0, c):
            if (grid[x][y] == '%'):
                what = 0
                cost = 100000000
            elif(grid[x][y] == '-'):
                what = 1
                cost = 1
            elif (grid[x][y] == '.'):
                what = 2
                cost = 0
            else:
                what = 3
            row.append(Cell(what, cost))
        cellGrid.append(row)
    return cellGrid

answered May 14, 2014 at 3:57

user764357

Sign up to request clarification or add additional context in comments.

1 Comment

David Over a year ago

Nice and simple. Thanks!

jspurim · Accepted Answer · 2014-05-14 04:02:19Z

1

Use list comprehensions to create a list of lists full of Nones. Change cellGrid initialization to:

cellGrid = [ [None for i in range(c)] for j in range(r)]

then you do

cellGrid[x][y] = Cell(what,cost)

Full code example:

def insertion (r, c, grid):
    cellGrid = [ [None for i in range(c)] for j in range(r)]
    for x in range(0, r):
        for y in range(0, c):
            if (grid[x][y] == '%'):
                what = 0
                cost = 100000000
            elif(grid[x][y] == '-'):
                what = 1
                cost = 1
            elif (grid[x][y] == '.'):
                what = 2
                cost = 0
            else:
                what = 3
            cellGrid[x][y] = Cell(what,cost)
    return cellGrid

edited May 14, 2014 at 4:02

answered May 14, 2014 at 3:55

jspurim

9558 silver badges25 bronze badges

Collectives™ on Stack Overflow

Creating list of lists containing objects in python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related