我有以下 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,因为求和并因此给我们所需的输出。
