如何将这个多维列表重塑为二维数组？

Question

我知道这个问题已经被问过几十次了，我试图获得使用多维数组的直觉（here，here），但我无法弄清楚这个过程。

我使用来自Calendar 库的yeardatescalendar() 获取一年中的几天、几周、几个月，并打包到一个4D 列表中。

import calendar

cal = calendar.Calendar()
yr_11 = cal.yeardatescalendar(2011)

返回值是月份行列表。每个月行最多包含 3 个月。每个月包含 4 到 6 周，每周包含 1 到 7 天。

我正在尝试将其转换为二维数组，因此它只是一个周列表。但即使我试图这样做，我也不理解输出。

# create an array
arr = np.array(yr_11)

arr.ndim # this returns '2'

arr.shape # this returns (4,3)

# yr_11 holds 63 weeks, so I tried to reshape
new_array = arr.reshape(63,1)

但它会抛出错误：ValueError: cannot reshape array of size 12 into shape (63,1)

有人能解释一下发生了什么并帮助将其转换为二维数组吗？

编辑：基本上我正在寻找这个

[[datetime.date(2010, 12, 27),
 datetime.date(2010, 12, 28),
 datetime.date(2010, 12, 29),
 datetime.date(2010, 12, 30),
 datetime.date(2010, 12, 31),
 datetime.date(2011, 1, 1),
 datetime.date(2011, 1, 2)],
[datetime.date(2011, 1, 3),
 datetime.date(2011, 1, 4),
 datetime.date(2011, 1, 5),
 datetime.date(2011, 1, 6),
 datetime.date(2011, 1, 7),
 datetime.date(2011, 1, 8),
 datetime.date(2011, 1, 9)], ... ]

Answer 1

yeardatescalendar 返回的嵌套list 具有以下层次结构：

• 多月块
• 月
• 周
• 天

因此，例如，yr_11[2][1][3][2] 将在第 3 块中给出代表第 2 个月第 4 周第 3 天的 datetime。请记住，索引是从左（嵌套最少）到右（嵌套最多）进行的，但我们通常以相反的方向读取这些元素（最细化的优先）

为了简化我们的计算，我们可以传递width=12，这样我们的结果将包含一个包含 12 个月的数据块。

接下来，只需将list 展平并将结果传递给np.array：

import calendar

cal = calendar.Calendar()
yr_11 = cal.yeardatescalendar(2011, width=12)

flat = [day for month in yr_11[0] for week in month for day in week]
dates = np.array(flat)

print(dates)

输出：

[datetime.date(2010, 12, 27) datetime.date(2010, 12, 28)
 datetime.date(2010, 12, 29) datetime.date(2010, 12, 30)
 datetime.date(2010, 12, 31) datetime.date(2011, 1, 1)
 ...
 datetime.date(2011, 12, 28) datetime.date(2011, 12, 29)
 datetime.date(2011, 12, 30) datetime.date(2011, 12, 31)
 datetime.date(2012, 1, 1)]