如何摆脱numpy数组中值周围的括号？ - Python

Question

我目前正在处理一些气候数据，但它的形状很奇怪。数组如下所示：

       ([[0.02115021]],

        [[0.03046454]],

        [[0.05636626]],

        [[0.08100581]],

        [[0.1113209 ]],

        [[0.11042633]],

        [[0.12332429]],

        [[0.1256145 ]],

        [[0.13792552]],

        [[0.11826107]],

        [[0.05710823]]],
  mask=False,
  fill_value=1e+20,
  dtype=float32)

但我只想要一个看起来像 ([1,2,3,4,5,6,7]) 的简单 numpy 数组，因为这是一个时间序列。我试图用 np.asarray(data) 转换它，但值周围的双括号仍然存在，这使得处理数据有点不可能。有人知道如何摆脱它们吗？

非常感谢。

Answer 1

numpy数组的Flatten方法可用于将n-d数组转换为1d数组。

a = np.array([[1,2],[3,4]])
a.flatten()
# output: array([1, 2, 3, 4])

更多信息见 https://docs.scipy.org/doc/numpy/reference/generated/numpy.array.html

Answer 2

这很容易使用，称为列表推导：

[wi[0][0] for wi in data]