Python 'NoneType' Error

[Mahem Ones]

I want to print out list pieces without any zeros to make it more readable.
Part of this code is in a while loop.

import copy
pieces = [ [1,2,3,4,5,6] , [0,0,0,0,0,0], [0,0,0,0,0,0] ]
def formatt (formatt) :
        while 0 in formatt[1] :
                formatt[1].remove(0)
        while 0 in formatt[2] :
                formatt[2].remove(0)
        while 0 in formatt[0] :
                formatt[0].remove(0)
        print formatt
 
        #part in loop \/
        form = copy.deepcopy(pieces)
        formatt(form)
 

But I getting this error message after the loop runs once:

Traceback (most recent call last):
File "C:\Documents and Settings\Jos\Desktop\Towers of Henoy.py", line 68, in <module>
formatt(form)
File "C:\Documents and Settings\Jos\Desktop\Towers of Henoy.py", line 37, in formatt
while 0 in formatt[1] :
TypeError: 'NoneType' object is not subscriptable

Can anyone help?


-Thanks
mahem1

[bvdet]

This can be accomplished more succinctly with a list comprehension:

>>> pieces = [ [1,2,3,4,5,6] , [0,0,0,0,0,0], [0,0,0,0,0,0] ]
>>> [[num for num in item if num != 0] for item in pieces]
[[1, 2, 3, 4, 5, 6], [], []]
>>> 

[Mahem Ones]

That works great!!
The error:
'TypeError: 'NoneType' object is not subscriptable' /iterable
Was because I used 'return pieces' instead of 'return (pieces)'

-Thanks
mahem1

[dwblas]

'TypeError: 'NoneType' object is not subscriptable' /iterable
Was because I used 'return pieces' instead of 'return (pieces)'

I suspect that it was because you use the same name for the function and the variable (which one are you referring to in your code). A variable is of type string, int, etc. A function is probably None type.

def formatt (formatt) :