Skip to main content
 首页 » 编程设计

python之根据另一个数组中的值扩展一个 numpy 数组

2024年08月12日31zdz8207

我有以下 numpy 数组 a = np.array([1,1,2,1,3]) 应该转换为以下数组 b = np.array ([1,1,1,1,1,1,1,1])。

a 数组中的所有非 1 值 都应在 b 数组中扩展为它们在一个数组。简单地说,2应该变成2,3应该变成3。

坦率地说,我找不到执行此操作的 numpy 函数,但我确定存在一个。非常欢迎任何建议!谢谢!

请您参考如下方法:

我们可以简单地做-

np.ones(a.sum(),dtype=int) 

这将容纳所有数字:1 和非 1,因为求和并因此给我们所需的输出。