numpy 搜索数组中的元素(分散和减少)



我正在尝试在此代码中将给定元素搜索到具有随机数的数组中。使用分散和减少,我只想计算元素在数组中存在的次数。我分散数组,搜索是在拆分子数组中完成的。每个拆分数组都会查找在此子数组中找到该元素的次数,然后使用 reduce i 收集在整个主数组中找到该元素的所有时间。


触发错误是因为recv_buf.shape(1,N//nprocs)

让它成为 1-Drecvbuf= np.empty ([N//nprocs], 'd'),它会没问题(注意我使用了//,所以你的程序适用于 py2 和 py3(。

顺便说一下,您可能需要查看以下内容以计算发生次数: 如何计算 Python 中某个项目在 ndarray 中的出现次数?

此外,您混合了不同的类型:进程零上的a没有类型化(所以它是 int64(,a其他进程上的类型是"i"(所以它是 int32,但你不使用它,所以这是一个真正的问题(,recvbuf 是"f"(float32(。

最新更新