Question

我有像p=array([ 1,2,3,4,5,6,7,8,9,10,11,12])

这样的1d numpy数组

在我的工作中，我需要将其更改为2d数组，如

p_reshape = array([[1,2,3], 
                   [4,5,6],
                   [6,7,8],
                   [9,10,11]])

我有其他函数返回p矩阵中元素i的位置（1d）说p（i）

现在我想知道p（i）在转换后的2d矩阵p_reshape中的位置。

下面显示了从1d转换为2d的代码：

row=300
col=500
size=row*col
p=np.ones((size))
p_reshape=np.reshape((row,size))

我希望我能很好地表达自己的想法。

Answer 1

我想你想要numpy.unravel_index

In [3]: import numpy as np
In [4]: p = np.array([ 1,2,3,4,5,6,7,8,9,10,11,12])

In [5]: p_reshape = p.reshape(4,3)

In [6]: p_reshape
Out[6]:
array([[ 1,  2,  3],
       [ 4,  5,  6],
       [ 7,  8,  9],
       [10, 11, 12]])

In [7]: np.unravel_index(5, p_reshape.shape)
Out[7]: (1, 2)

Answer 2

如果我理解正确，你可以使用：

p_reshape[i // ncols, i % ncols]

不能吗？

p = np.arange(1,13)
p_reshape = p.reshape((4,3))
ncols = p_reshape.shape[1]
all(p_reshape[i // ncols, i % ncols] == p[i] for i in range(p.size))

True

如何从1D重塑为2D numpy数组后返回元素的位置（索引）？

2 个答案: