如何从生成器构建numpy数组?

如何从生成器构建numpy数组?,第1张

如何从生成器构建numpy数组

Numpy数组要求在创建时明确设置其长度,这与python列表不同。这是必要的,以便可以在内存中连续分配每个项目的空间。连续分配是numpy数组的关键特性:此方法与本机代码实现相结合,使对它们的 *** 作比常规列表执行得快得多。

请牢记这一点,从技术上讲,不可能采用生成器对象并将其转换为数组,除非您执行以下任一 *** 作:

  1. 可以预测运行时将产生多少个元素:

    my_array = numpy.empty(predict_length())

    for i, el in enumerate(gimme()): my_array[i] = el

  2. 愿意将其元素存储在中间列表中:

    my_array = numpy.array(list(gimme()))
  3. 可以制作两个相同的生成器,遍历第一个生成器以找到总长度,初始化数组,然后再次遍历生成器以查找每个元素:

    length = sum(1 for el in gimme())

    my_array = numpy.empty(length)
    for i, el in enumerate(gimme()): my_array[i] = el

1 可能是您要寻找的。 2 是空间效率低下的,而 3 是时间效率低下的(您必须两次通过生成器)。



欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5631155.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-15
下一篇 2022-12-15

发表评论

登录后才能评论

评论列表(0条)

保存