Numpy 数组理解

Question

有没有办法在Python中进行numpy数组理解？我看到它的唯一方法是使用列表理解，然后将结果转换为 numpy 数组，例如np.array(list comprehension). 我本来期望有一种方法可以直接使用 numpy 数组来完成此操作，而不使用列表作为中间步骤。

另外，是否可以重载列表运算符，即[和]，以便结果是 numpy 数组，而不是列表。

Answer 1

您可以从生成器表达式创建 numpy 数组。您只需提前指定 dtype：

import numpy as np
x = np.fromiter(range(5), dtype=int)
y = np.fromiter((i**2 for i in range(5)), dtype=int)