Unexpected behavior in nested for loop in python

Question

Very simple code nested example:

All the code does is create a list of lists that is initialized to zero. It iterates through the list rows and columns and each position is given a value. For some reason the last row of the 2D list is duplicated for each row when the final vector is printed.

Number_of_channels=2
Coefficients_per_channel=3

coefficient_array=[[0]*Coefficients_per_channel]*Number_of_channels 
print coefficient_array

for channel in range(Number_of_channels):
    for coeff in range(Coefficients_per_channel):
        coefficient_array[channel][coeff]=coeff*channel
        print coefficient_array[channel][coeff]

print coefficient_array

Output:

[[0, 0, 0], [0, 0, 0]]
0
0
0
0
1
2
[[0, 1, 2], [0, 1, 2]]

I actually expect:

[[0, 0, 0], [0, 1, 2]]

Anyone have any idea how come this is happening?

Martijn Pieters · Accepted Answer · 2012-05-31 09:57:33Z

5

You only duplicate the outer list, but the values of that list are left untouched. Thus, all (both) outer lists contain references to the same inner, mutable list.

>>> example = [[1, 2, 3]]
>>> example *= 2
>>> example
[[1, 2, 3], [1, 2, 3]]
>>> example[0][0] = 5
[[5, 2, 3], [5, 2, 3]]
>>> example[0] is example[1]
True

Better create the inner lists in a loop:

coefficient_array=[[0]*Coefficients_per_channel for i in xrange(Number_of_channels)]

or, illustrated with the python prompt again:

>>> example = [[i, i, i] for i in xrange(2)]
>>> example
[[0, 0, 0], [1, 1, 1]]
>>> example[0][0] = 5
>>> example
[[5, 0, 0], [1, 1, 1]]
>>> example[0] is example[1]
False

edited May 31, 2012 at 9:57

answered May 31, 2012 at 9:26

Martijn Pieters

1.1m326 gold badges4.2k silver badges3.4k bronze badges

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

1 Comment

stanri Over a year ago

Thanks for this, I understand now. Holy crap, python is powerful.

glglgl · Accepted Answer · 2012-05-31 09:39:53Z

1

With

coefficient_array=[[0]*Coefficients_per_channel]*Number_of_channels

you do a duplication of references to the same object:

coefficient_array[0] is coefficient_array[1]

evaluates to True.

Instead, build your array with

[[coeff*channel for coeff in range(Coefficients_per_channel)] for channel in range(Number_of_channels)]

answered May 31, 2012 at 9:39

glglgl

91.5k13 gold badges157 silver badges230 bronze badges

Comments

Maria Zverina · Accepted Answer · 2012-05-31 09:31:08Z

0

Try this instead:

coefficient_array=[0]*Number_of_channels 
print coefficient_array

for channel in range(Number_of_channels):
    coefficient_array[channel] = [0] * Coefficients_per_channel
    for coeff in range(Coefficients_per_channel):
        coefficient_array[channel][coeff]=coeff*channel
        print (channel, coeff)
        print coefficient_array[channel][coeff]

print coefficient_array

answered May 31, 2012 at 9:31

Maria Zverina

11.2k3 gold badges47 silver badges62 bronze badges

1 Comment

Bittrance Over a year ago

How about replacing the inner loop with coefficient_array[channel] = range(0, Coefficients_per_channel * channel, channel)?

Collectives™ on Stack Overflow

Unexpected behavior in nested for loop in python

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related