How do you print out elements from a Numpy array on new lines using a for loop?

Question

Create an array with numpy and add elements to it. After you do this, print out all its elements on new lines.

I used the reshape function instead of a for loop. However, I know this would create problems in the long run if I changed my array values.

import numpy as np

a = np.array([0,5,69,5,1])

print(a.reshape(5,1))

How can I make this better? I think a for loop would be best in the long run but how would I implement it?

Valdi_Bo · Accepted Answer · 2020-11-14 10:23:59Z

2

Some options to print an array "vertically" are:

print(a.reshape(-1, 1)) - You can pass -1 as one dimension, meaning "expand this dimension to the needed extent".
print(np.expand_dims(a, axis=1)) - Add an extra dimension, at the second place, so that each row will have a single item. Then print.
print(a[:, None]) - Yet another way of reshaping the array.

Or if you want to print just elements of a 1-D array in a column, without any surrounding brackets, run just:

for x in a:
    print(x)

answered Nov 14, 2020 at 10:23

Valdi_Bo

31.1k4 gold badges29 silver badges45 bronze badges

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

Comments

Ali Mizan · Accepted Answer · 2020-11-13 22:54:48Z

0

You could do this:

print(a.reshape([a.shape[0], 1]))

This will work regardless of how many numbers are in your numpy array.

Alternatively, you could also do this:

[print(number) for number in a.tolist()]

answered Nov 13, 2020 at 22:47

Ali Mizan

1,9201 gold badge22 silver badges20 bronze badges

If you want to change from a row array to a column array better use a.reshape(-1,1)