python - 小时 (24) 和天 (365) 的 numpy 矩阵

Question

我有两个向量；一个代表一天中的几个小时 [1,2,3,...,24]，第二个代表一天中的几天年 [1,2,3,4,5,6,...,365]

我想构造一个 24*365 单元格、24 行和 365 列的矩阵。

类似于:

a = [(1,24),(2,24),(3,24),(4,24),(5,24),...,(365,24),
     (1,23),(2,23),(3,23),(4,23),(5,23),...,(365,23),
     (1,22),(2,22),(3,22),(4,22),(5,22),...,(365,22),
          .,
          .,
          .,
          .,
     (1,1),(2,1),(3,1),(4,1),(5,1),...,(365,1)]

在我想应用函数 f(x,y) 并将 x,y 替换为 z 之后(保持不变矩阵结构)。最终这将被转换为带有渐变的彩色 map 。

Answer 1

可能值得注意的是，虽然您可以将通用对象存储在 numpy 数组中，但这可能不是一个好主意 - 大多数算法都经过优化以在矩阵中的每个槽中具有单个值。

这样做的结果是，您可能不会以每个槽中有两个值的 24 x 365 元素矩阵结束，而是以 2 x 24 x 365 元素的 numpy 数组结束。

执行类似操作的一种方法是:

hours = numpy.arange(365).reshape(1,-1).repeat(24,axis=0)
days = numpy.arange(24).reshape(-1,1).repeat(365,axis=1)
full = numpy.array([days, hours])
print full
print full.shape

给出:

[[[  0   0   0 ...,   0   0   0]
  [  1   1   1 ...,   1   1   1]
  [  2   2   2 ...,   2   2   2]
  ..., 
  [ 21  21  21 ...,  21  21  21]
  [ 22  22  22 ...,  22  22  22]
  [ 23  23  23 ...,  23  23  23]]

 [[  0   1   2 ..., 362 363 364]
  [  0   1   2 ..., 362 363 364]
  [  0   1   2 ..., 362 363 364]
  ..., 
  [  0   1   2 ..., 362 363 364]
  [  0   1   2 ..., 362 363 364]
  [  0   1   2 ..., 362 363 364]]]
(2, 24, 365)