python - 在 Python 中从现有 2D 数组创建 3D 数组

Question

我正在尝试创建一个具有尺寸的数组:

a(Days,Hours,Station)

我有 61 天内 2 个站点的数组“a”的每小时数据，因此目前我有以下维度的数组:

a(1464,2)

其中 1464 是我拥有的每个站点每小时数据点的数量(24 小时*61 天)。然而，我想进一步分解它并添加另一个包含天的维度，因此维度将是:

a(61 days,24 hours/day, 2 stations)

关于如何正确地获取我当前拥有的数组“a”并将其更改为这 3 个维度，有什么想法吗？

Answer 1

这会将数组 a 分割为最大长度 size 的 block 。

def chunks( a, size ):
    arr = iter( a )
    for v in arr:
        tmp = [ v ]
        for i,v in zip( range( size - 1 ), arr ):
            tmp.append( v )

        yield tmp

splitted = list( chunks( a, 24 ) )