如何在 Python 中索引 0-d 数组？答案

【问题标题】：How to index 0-d array in Python?如何在 Python 中索引 0-d 数组？
【发布时间】：2013-09-01 17:28:15
【问题描述】：

这可能是一些常见问题解答中存储的众所周知的问题，但我无法通过谷歌搜索解决方案。我正在尝试编写标量参数的标量函数，但允许使用 ndarray 参数。该函数应检查其参数的域正确性，因为域冲突可能会导致异常。这个例子展示了我尝试做的事情：

import numpy as np
def f(x):
    x = np.asarray(x)
    y = np.zeros_like(x)
    y[x>0.0] = 1.0/x
    return y

print f(1.0)

在分配y[x>0.0]=... 时，python 说0-d arrays can't be indexed。解决此执行的正确方法是什么？

【问题讨论】：

【解决方案1】：

这将在 NumPy >= 1.9 中正常工作（在撰写本文时尚未发布）。在以前的版本中，您可以通过额外的np.asarray 电话解决：

x[np.asarray(x > 0)] = 0

【讨论】：

【解决方案2】：

你可以打电话给f([1.0])吗？

否则你可以这样做：

x = np.asarray(x)
if x.ndim == 0:
    x = x[..., None]

【讨论】：