Creating an array of arrays dynamically in Python

Question

I would like to create the following array of arrays:

[[1, 1, 1],
 [1, 2, 4],
 [1, 3, 9],
 [1, 4, 16],
 [1, 5, 25],
 [1, 6, 36],
 [1, 7, 49],
 [1, 8, 64],
 [1, 9, 81],
 [1, 10, 100]]

And this is my code:

rows = 10
cols = 3
arr = []
x = [1, 1, 1]

for i in range(rows) :

    for j in range(cols) :
        x[j] = (i + 1) ** j

    arr.append(x)

print(arr)

The output I'm getting is:

[[1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100],
 [1, 10, 100]]

It seems like there's an error somewhere, but I cant see it, any feedback would be appreciated!

Add print statements for I, j, x[j] and i+1**j. Do you see what's happening? — 2e0byo
– 2e0byo, Commented Nov 10, 2021 at 10:31

bugra · Accepted Answer · 2021-11-10 10:32:48Z

1

You should not use the same x array in every loop but create it from scratch:

rows = 10
cols = 3
arr = []
for i in range(rows):
    x = []
    for j in range(cols):
        x.append((i + 1) ** j)
    arr.append(x)
print(arr)

answered Nov 10, 2021 at 10:32

bugra

1293 bronze badges

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

Comments

Andrea Barnabò · Accepted Answer · 2021-11-10 10:30:10Z

1

This can help you ?

rows = 10
arr = [[1, i, i*i] for i in range(1,rows + 1)]
print(arr)

answered Nov 10, 2021 at 10:30

Andrea Barnabò

5746 silver badges19 bronze badges

Comments

mozway · Accepted Answer · 2021-11-10 10:48:23Z

1

A variation on @Andrea's solution. You can think it in terms of powers:

i**0 -> 0 ; i**1 -> 1 ; i**2 -> i*i, etc.

rows = 10
cols = 3
[[(i+1)**n for n in range(cols)] for i in range(rows)]

answered Nov 10, 2021 at 10:48

mozway

267k13 gold badges56 silver badges106 bronze badges

Collectives™ on Stack Overflow

Creating an array of arrays dynamically in Python

3 Answers 3

Comments

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Related