Duplicate every value in array as a new array

Question

I have an numpy ndarray input like this

[[T, T, T, F],
  [F, F, T, F]]

and I want to duplicate every value as a new array, so the output would be

[[[T,T], [T,T], [T,T], [F,F]]
  [[F,F], [F,F], [T,T], [F,F]]]

How can I do this? Thank you in advance

Yeah but I was wondering if there was a one line method without using a loop. — ErgiS
– ErgiS, Commented Jul 9, 2019 at 10:25

yatu · Accepted Answer · 2019-07-09 10:30:04Z

4

One way would be using np.dstack to replicate the array along the third axis:

np.dstack([a, a])

array([[['T', 'T'],
        ['T', 'T'],
        ['T', 'T'],
        ['F', 'F']],

       [['F', 'F'],
        ['F', 'F'],
        ['T', 'T'],
        ['F', 'F']]], dtype='<U1')

Setup:

T = 'T'
F = 'F'
a = np.array([[T, T, T, F],
              [F, F, T, F] ])

edited Jul 9, 2019 at 10:30

answered Jul 9, 2019 at 10:26

yatu

88.6k12 gold badges93 silver badges148 bronze badges

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

Comments

ncica · Accepted Answer · 2019-07-09 10:29:57Z

0

you can just use list comprehension:

data =[['T', 'T', 'T', 'F'],
  ['F', 'F', 'T', 'F'] ]

d = [[[i]*2 for i in j] for j in data]
print (d)

output:

[[['T', 'T'], ['T', 'T'], ['T', 'T'], ['F', 'F']], [['F', 'F'], ['F', 'F'], ['T', 'T'], ['F', 'F']]]

answered Jul 9, 2019 at 10:29

ncica

7,2161 gold badge19 silver badges41 bronze badges

Collectives™ on Stack Overflow

Duplicate every value in array as a new array

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related