索引Numpy数组(4维)

时间:2016-04-07 08:17:09

标签: python arrays numpy matrix indexing

我有一个numpy数组,它是(1,2048,1,1)。我需要将前两个维度分配给另一个numpy数组(1,2048),但我对如何正确索引感到困惑。希望你能帮忙!

2 个答案:

答案 0 :(得分:0)

我通过使用np.squeeze(x)来删除单例维度来解决它。

答案 1 :(得分:0)

在最后两个维度为1,1的特定情况下,您还可以使用reshape

b = a.reshape((1,2048))