检查numpy矩阵中特定元素的值

【问题标题】：check value of a specific element in numpy matrix检查numpy矩阵中特定元素的值
【发布时间】：2020-04-08 05:42:20
【问题描述】：

i 使用 if else 来检查 numpy 矩阵的索引 i,i 处的元素是否为 0

if a[i, i]==0:

得到错误 ValueError：具有多个元素的数组的真值不明确。使用 a.any() 或 a.all() 有什么解决方法吗？

【问题讨论】：

【解决方案1】：

您的代码的问题在于 a[i, i] 不是单个值，而是列表或 numpy 数组。而且您不能仅使用一个 if condition 来检查列表/numpy 数组中的所有值。
你可以做的是：

for item in a.flatten():
    if item == =.
       # do sth

【讨论】：