从数组中删除小数

Question

我有两个数组包含0和1。我想对它们执行hstack（），但无法获得所需的输出。在

Python代码。。在

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

电流输出。。在

^{pr2}$

预期输出。。在

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

我不明白我在犯什么愚蠢的错误。在

Answer 1

必须告诉numpy以整数形式返回值

import numpy as np

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

像这样打印zerosThenOnes，[0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1]

使用：

^{pr2}$

Numpy Zeros

Answer 2

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

对于np.array输出，或

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

对于list输出