码:
n=3
x=np.fromfunction(lambda i,j: (i==1)and(j==1), (n,n), dtype=int)
导致“ ValueError:具有多个元素的数组的真值不明确。请使用a.any()或a.all()”
有什么问题?
最佳答案
该文档具有误导性。不会使用每个单元格的索引重复调用该函数;它被调用一次,索引数组一次代表所有单元格的索引。这个函数调用的返回值直接返回:
>>> numpy.fromfunction(lambda *args: 1, (2, 2))
1
>>> numpy.fromfunction(lambda *args: args, (2, 2))
(array([[ 0., 0.],
[ 1., 1.]]), array([[ 0., 1.],
[ 0., 1.]]))
您需要更改功能以这种方式操作:
lambda i, j: (i==1) & (j==1)
# ^ elementwise bitwise and