python - 确保变量是数组，无论它是列表还是标量

Question

我经常发现我需要确保它x是一个类似数组的对象，无论它是作为浮点数还是作为列表出现。

我最终需要 numpy 数组，所以我希望 np.array() 可能是一个简单的解决方案。但实际上括号仍然是一个问题。

我想出的最佳解决方案是

def EnsureArray(x):
    if np.isscalar(x):
        return np.array([x])
    else:
        return np.array(np.x)

可以吗，还是有更好的东西（没有定义我自己的功能？）

标签： pythonarraysnumpy