将3d矩阵转换为特征向量

时间:2014-09-08 06:02:53

标签: python numpy scikit-learn

所以我有数据形状(100,100,5000)。

基本上,它是一个100乘100像素的图像,每个(x,y)像素都有一些光谱矢量。 所以,数据是格式

     [ [ [ 0, 0.2.....],[0.1,0.3.....].. and so on]]

当我们做d [0] [0] [0]对应于图像中的(0,0)像素且光谱值为0

现在,我想在其中运行kmeans算法。这就是背景。

要运行kmeans,我想将其转换为特征向量..作为

[0,0,0]
[0,0,0.2]
and so on..

是否有" python"这种转换的方式。我猜应该有一些numpy reshape方法可以很容易地实现这一点,而不是编写我试图避免的三个for循环。

1 个答案:

答案 0 :(得分:2)

使用numpy reshape http://docs.scipy.org/doc/numpy/reference/generated/numpy.reshape.html例如

a.reshape((m,n))

其中mn是数组所需的坐标a