如何在numpy中向量化linspace

Question

我想对数组做linspace.如下:

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

我希望回来看起来像:

b = [[0, 0.5, 1, 1.5, 2]
     [0, 1, 2, 3, 4]
     [0, 1.5, 3, 4.5, 6]]

这是我的代码:

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)