如何合并Numpy数组的最后2个维度

时间:2019-03-01 04:20:50

标签: python-3.x numpy numpy-ndarray

我已经看到How to combine dimensions in numpy array?

尝试过,但是当我尝试

imgs.reshape(img.shape[:-2]+(-1))

我以为是逻辑扩展名,但出现错误:

can only concatenate tuple (not "int") to tuple

我打算对链接的问题的已接受答案发表评论,并直接向他们寻求帮助,而不是为此创建新线程,但是我没有必要的信誉来发表评论。

编辑: 例如,我有一个形状为(x,y,8,8)的输入数组,其中x和y可以根据输入到函数中的图像进行更改。我想将其制成形状为(x,y,64)的阵列。在这种情况下,64永远不会改变。

1 个答案:

答案 0 :(得分:2)

您必须添加逗号才能将其识别为元组。试试

imgs.reshape(img.shape[:-2] + (-1,))

或者,使用打开包装:

imgs.reshape((*img.shape[:-2], -1))