CODEWITHSUNDEEP
pythonnumpymatrixmultidimensional-array
Lucy
Lucy

Reputation: 1852

Remove Decimals from Array

I have 2 arrays containing zeros & ones. I want to perform hstack() on them but not getting the desired output.

Python Code..

 import numpy as np
 zeros = np.zeros(8)
 ones = np.ones(8)
 zerosThenOnes = np.hstack((zeros, ones))   # A 1 by 16 array

Current Output..

 [ 0.  0.  0.  0.  0.  0.  0.  0.  1.  1.  1.  1.  1.  1.  1.  1.]

Expected Output..

 [ 0   0   0   0   0   0   0   0   1   1   1   1   1   1   1   1 ]

I can't understand what silly mistake I'm doing.

Upvotes: 0

Views: 10506

Answers (2)

dermen
dermen

Reputation: 5372

np.hstack((np.zeros(8), np.ones(8))).astype(int)

for np.array output, or

map( int, np.hstack((np.zeros(8), np.ones(8))) )

for list output

Upvotes: 0

Dobz
Dobz

Reputation: 1213

You must tell numpy to return the values as integers

import numpy as np

zeros = np.zeros((8,), dtype=np.int)
ones = np.ones((8,), dtype=np.int)
zerosThenOnes = np.hstack((zeros, ones))

To print out zerosThenOnes like this [0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1]

Use:

print([x for x in zerosThenOnes])

Numpy Zeros

Upvotes: 1

Related Questions