我想你是假设你的函数被调用了n * n次，每次都有一对新的可能的索引0... n。试着打印这些值来看看发生了什么：

>>> import numpy as np

>>> n = 3
>>> def f(i, j):
...     print(i)
...     print(j)
...     return n if i==j else (i+1) + (j+1)

>>> np.fromfunction(f, (n,n), dtype=int)
[[0 0 0]
 [1 1 1]
 [2 2 2]]
[[0 1 2]
 [0 1 2]
 [0 1 2]]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "...\Python\Python311\site-packages\numpy\core\numeric.py", line 1866, in fromfunction
    return function(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "<stdin>", line 4, in f
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

相反，该函数只被调用一次，并使用可能索引的网格！
在... if i == j else ...部分中，在比较了i和j之后，您将得到一个具有True和False值的布尔数组，因此numpy不确定该怎么做。
相反，对于这种if-else逻辑，依赖于np.where，并且可能使用一点广播来提高效率：

>>> xs, ys = np.indices((n, n), sparse=True)
>>> np.where(np.eye(n), n, xs + ys + 2)
array([[5, 3, 4, 5, 6],
       [3, 5, 5, 6, 7],
       [4, 5, 5, 7, 8],
       [5, 6, 7, 5, 9],
       [6, 7, 8, 9, 5]])