Python - adding string elements from a list into an array

Question

Here's an example of what I want to do. I want to take b and add this to my array but in a particular format (again I just want to know the steps to do this, my code deals with strings):

import numpy as np

a = np.zeros(25, dtype= np.character).reshape(5,5)
b = [1,2,3,4,5,6,7,8,9]

for i in b:

    a[1:4,1:4] = i

print a

Output:

[['' '' '' '' '']
 ['' '9' '9' '9' '']
 ['' '9' '9' '9' '']
 ['' '9' '9' '9' '']
 ['' '' '' '' '']]

But what I want is this:

[['' '' '' '' '']
 ['' '1' '2' '3' '']
 ['' '4' '5' '6' '']
 ['' '7' '8' '9' '']
 ['' '' '' '' '']]

Could anyone give me an idea on how to do this? Thank you.

unutbu · Accepted Answer · 2014-03-20 01:01:26Z

2

import numpy as np

a = np.zeros(25, dtype= np.character).reshape(5,5)
b = np.array([1,2,3,4,5,6,7,8,9])
a[1:4,1:4] = b.reshape(3,3)
print(a)

yields

[['' '' '' '' '']
 ['' '1' '2' '3' '']
 ['' '4' '5' '6' '']
 ['' '7' '8' '9' '']
 ['' '' '' '' '']]

answered Mar 20, 2014 at 1:01

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

Collectives™ on Stack Overflow

Python - adding string elements from a list into an array

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related