如何在numpy中矢量化linspace

Question

我想对数组执行linspace操作。如下所示：

a = np.array([2, 4, 6])
b = vectorize(np.array)(0, a, 5)

我想要一些看起来像：

^{pr2}$

这是我的代码：

import numpy as np
a = np.arange(1001)
c = np.vectorize(np.linspace)(0, a, 101)
print(c)

它显示：ValueError:使用序列设置数组元素。 在没有for循环的numpy中有什么方法可以做到这一点吗？在

Answer 1

建立自己的：

def vlinspace(a, b, N, endpoint=True):
    a, b = np.asanyarray(a), np.asanyarray(b)
    return a[..., None] + (b-a)[..., None]/(N-endpoint) * np.arange(N)