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)

如何在numpy中向量化linspace

1 个答案: