python - 有一些 numpy.map 吗？

Question

我可能在这里遗漏了一些明显的东西，但我遗漏了一个函数 numpy.map。这与 Python 的 map 函数相同，但将输出收集在 numpy 数组中。例如，我可以有一个图像生成器 genImage(i)，它基于单个输入生成一个二维图像(大小为 (m, n))，我会喜欢将 range(k) 输入到我的生成器函数并获得一个 (k, m, n) 数组。

目前，我会使用 numpy.array(list(map(genImage, range(k)))，但我觉得这种转换成列表的效率很低(我的最终数组大约是50 GB)。因此，我正在寻找 numpy.map(genImage, range(k))，它类似于 numpy.fromiter，但用于多维输出迭代器。

(我试过 np.array(map(...))，但它返回一个包含 map 的单元素数组，因为它是唯一的条目 - 这就是原因:Why is it required to typecast a map into a list to assign it to a pandas series? )

有没有更好的方法来实现我想要的？我正在寻找一种理想的方式，我可以使用 joblib。

Answer 1

如果我没猜错，你需要 column_stack那会像这样工作:

import numpy as np

a = np.array([[1, 2], [3, 4], [5, 6]])

a = np.column_stack((range(3), a))

a
[[0 1 2]
 [1 3 4]
 [2 5 6]]