从数组中删除小数

Question

我有2个包含零和一的数组。我想对它们执行 hstack((，但没有得到所需的输出。

蟒蛇代码..

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

电流输出..

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

预期输出..

 [ 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]

用：

print([x for x in zerosThenOnes])

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输出