在Python中关联两个3D数组

时间:2015-03-04 18:22:57

标签: python arrays numpy

我需要在Python中关联两个数据集。一个数组是.mat文件,另一个是.bin文件列表。从这些数据集中,我创建了两个具有相同范围的3D阵列(120x112x244)。虽然熟悉Python,但我之前没有使用过这样的数据集,因此我正在寻求关于如何关联这些数组的建议。我尝试numpy相关并收到: “ValueError:对象太深,不适合所需的数组”

任何建议都将不胜感激

1 个答案:

答案 0 :(得分:4)

我想尝试的一个想法是首先展平3D矩阵,然后使用coorelate - 因为coorelate只需要1D向量。

http://docs.scipy.org/doc/numpy/reference/generated/numpy.correlate.html

让我们说你的两个基质被称为AB

>>> import numpy
>>> array_a = numpy.ndarray.flatten(A)
>>> array_b = numpy.ndarray.flatten(B)
>>> results = numpy.correlate(array_a, array_b)