将没有双循环的numpy数组展平

时间:2016-06-28 10:33:45

标签: python numpy

我有一个二维矩阵。就本例而言,假设它是一个随机矩阵

>>> a = np.random.randn(5, 7)
>>> a
array([[-0.37279322,  0.28619523, -0.05309901,  0.26010327,  0.1846693 , 0.33112176,  0.75814911],
       [ 1.57001151, -0.86831693, -0.20576395,  1.46450855, -0.01631132, 3.02790403, -0.65313017],
       [ 0.2362675 , -1.52190536,  0.04687194,  2.01618876,  0.03780218, -0.53041096, -0.30104844],
       [-0.5504834 ,  1.04286156,  1.12863785,  0.89583492,  0.28607363, 1.42858007,  0.28582572],
       [-0.768464  ,  0.31952554,  0.81129581,  0.26239668, -0.23242878, -1.01584339,  0.39573906]])

和两个标签向量:

label_y = np.array([23, 984, 123, 9321, 121238])
label_x = np.array([121, 31312, 9123131, 1111, 1231441, 1929313, 192312312361])

我想展平a的元素并输出它们的标签indeces和值。例如:

23,121,-0.37279322 
23,31312,0.28619523 
23,9123131,-0.05309901 
23,1111,0.26010327
23,1231441,0.1846693
23,1929313,0.33112176
23,192312312361,0.75814911 
984,121,...
...

有一种简单的方法可以在没有for循环的情况下进行numpy吗?

1 个答案:

答案 0 :(得分:4)

使用np.meshgrid创建与2DX标签对应的Y网格,然后将它们与2D输入数组{{1}一起堆叠为列},像这样 -

a