4

我想将存储在 1D 中的数据更改为 2D：我的意思是：

从

x|y|a
1|1|a(1,1)
2|1|a(2,1)
3|1|a(3,1)
1|2|a(1,2)
...

进入：

x\y|1     |2     |3
1  |a(1,1)|a(1,2)|a(1,3
2  |a(2,1)|a(2,2)|a(2,3)...
3  |a(3,1)|a(3,2)|a(3,3)...
    ...

我做了2个循环：

(rows - array of x,y,a)
for n in range(len(rows)):
            for k in range(x_len):
                    for l in range(y_len):
                            if ((a[2, n] == x[0, k]) and (a[3, n] == y[0, l])):
                                    c[k, l] = a[0, n]

但这需要很长时间，所以我的问题是在 Python 中是否有一个聪明而快速的解决方案。

所以要澄清我想要做什么：

我知道 return() 函数，关键是它在数组a中是随机的。

所以：

a = np.empty([4, len(rows)]

我将数据读入数据库中的数组a中，该数组有4 列 (1,2,x,y)和'len(rows)' 行。

我对“1”列感兴趣-我想将其放入新修改的数组中。

x = np.zeros([1, x_len], float)

y = np.zeros([1, y_len], float)

x是来自数组a的已排序列 (x)的向量，但没有长度为x_len的副本

（我通过 sql 查询读取它：select distinct ... ）

y是长度为y_len的数组a （不带副本）中的已排序列 (y)的向量

然后我正在制作数组：

c = np.zeros([x_len, y_len], float)

并将数组 a 中的数据放入 3 个循环（抱歉之前的错误）：>

 for n in range(len(rows)):
     for k in range(x_len):
        for l in range(y_len):
          if ((a[2, n] == x[0, k]) and (a[3, n] == y[0, l])):
              c[k, l] = a[0, n]

例子：

数组a

   array([[1, 3, 6, 5, 6], 
          [1, 2, 5, 5, 6], 
          [1, 4, 7, 1, 2],   ## x
          [2, 5, 3, 3, 4]])  ## y

向量：x和y

 [[1,2,4,7]]   ## x with x_len=4

 [[2,3,4,5]]   ## y with y_len=4

数组c

   array([[1, 5, 0, 0], 
          [0, 0, 0, 0],  
          [0, 0, 0, 3],  
          [0, 6, 0, 0]])

最后一个数组c看起来像这样（第一个 a[0] 被写入）：

 x\y 2|3|4|5
 -----------
 1   1|5|0|0
 2   0|0|0|0
 4   0|0|0|3
 7   0|6|0|0

我希望我没有弄错它是如何写入数组c的。

非常感谢您的帮助。

4

1 回答 1

6

你可以使用numpy：

>>> import numpy as np
>>> a = np.arange(9)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> a.reshape(3,3)
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
#or:
>>> a.reshape(3,3).transpose()
array([[0, 3, 6],
       [1, 4, 7],
       [2, 5, 8]])
于 2012-06-18T20:04:24.603 回答