numpy如何对一个数组中第一个为0的元素进行赋值

numpy如何对一个数组中第一个为0的元素进行赋值

1 个回答

小小的芭比

提问:numpy如何对一个数组中第一个为0的元素进行赋值

补充 : [2 0 4 0 6]将第一个0赋值为1

网友回答:

简单的代码参考

import numpy as np a=np.array([2,0,4,0,6]) a[np.argwhere(a==0)[0]]=1 print(a)

结果

[2 1 4 0 6]

  • 1 关注
  • 0 收藏, 浏览
  • 提出于

相似问题