Python中如何用单行代码为数组指定行添加特定元素值

• 单行实现代码：

  a[[5, 7], :] += a[5, 5]
  

• 原代码问题解析：

  • 你用的for循环会遍历a[(5,7),:]对应的两行数据，每一次循环都对这两行整体执行赋值操作，第一次循环给两行元素加了一次a[5,5]，第二次循环又重复加了一次，所以最终元素被添加了两次值。
  • numpy自带广播机制，支持直接对选中的多行进行批量算术运算，无需编写循环，一行代码就能完成需求。

内容的提问来源于stack exchange，提问作者421543