python - 如何向python函数传递多个参数

Question

我有一个包含 300 个条目的 numpy 数组。每个条目中都有另一个 numpy 数组 [2048 x 2048]。

每个条目都是一个“tiff”条目(矩阵形式)，对应于探测器中的像素位置。现在，我想做的是将其集中化，这样我就有一个 [2048 x 2048] 数组，其中每个条目都有 300 个条目，对应于 300 帧中的像素。

我想我已经找到了使用zip函数的方法。但是，每次我返回 [300 x 2048 x 2048] 或 [2048 x 300 x 2048]。

我想要一个 [2048 x 2048 x 300]。我正在尝试以一种相当经济和Pythonic的方式来做到这一点，而不仅仅是简单地重新加载到新数组并重新索引。

T_prime = zip(([list(t) for t in zip(*Tiffs)])) 

其中 Tiffs 是如上所述的数组。

Answer 1

在 numpy 中，我们经常向数组添加维度，而不是使用嵌套数组(这是列表示例的常态)。一旦将所有数据放入一个数组中，就可以轻松对其进行操作。就您而言，您似乎正在寻找转置数组。一个例子:

import numpy as np
example_data = np.empty(30, dtype=object)
for i in range(30):
    example_data[i] = np.zeros((100, 101))

structured = np.array(list(example_data))

print structured.shape
# (30, 100, 101)
print structured.transpose([1, 2, 0]).shape
# (100, 101, 30)