我试图用numpy hstack水平堆叠数组,但无法使其工作取而代之的是,它们都在一个列表中出现,而不是一个“矩阵式”的二维数组。

import numpy as np
y = np.array([0,2,-6,4,1])
y_bool = y > 0
y_bool = [1 if l == True else 0 for l in y_bool] #convert to decimals for classification
y_range = range(0,len(y))
print y
print y_bool
print y_range
print np.hstack((y,y_bool,y_range))

打印此:
[ 0  2 -6  4  1]
[0, 1, 0, 1, 1]
[0, 1, 2, 3, 4]
[ 0  2 -6  4  1  0  1  0  1  1  0  1  2  3  4]

我该如何让最后一行看起来像这样:
[0 0 0
 2 1 1
-6 0 2
 4 1 3]

最佳答案

如果要创建二维阵列,请执行以下操作:

print np.transpose(np.array((y, y_bool, y_range)))
# [[ 0  0  0]
#  [ 2  1  1]
#  [-6  0  2]
#  [ 4  1  3]
#  [ 1  1  4]]

10-07 14:48
查看更多