如何在ndarray中查找指定元素行号？代码报错ValueError求解

代码报错的核心原因 #

你写的代码里，list.index()方法在找不到目标元素时，不会返回None，而是直接抛出ValueError异常，所以当遍历第一行[0,1]时，执行array[i].index(2)就直接报错中断了，根本走不到后续的if判断逻辑。

for循环与if语句结合的逻辑原理 #

for循环的执行流程是：按顺序遍历迭代对象中的每个元素，每次迭代都会执行循环体内的代码，直到遍历完所有元素，或者遇到break语句提前终止。

if语句是条件分支判断：当条件为真时，执行if块内的代码；条件为假时，若有else分支则执行else块，否则跳过。

你原本的思路是“找不到就继续循环，找到就停止”，但忽略了index()方法的异常特性，导致逻辑无法正常执行。

修正后的代码（普通列表场景） #

用try-except捕获index()的异常，实现你的预期逻辑：

array = [[0,1],
         [2,3]]
item = 2
row_idx = -1
col_idx = -1

for i in range(len(array)):
    try:
        # 尝试在当前行找元素，找到返回列号
        col_idx = array[i].index(item)
        row_idx = i
        # 找到后直接跳出循环，不用继续遍历
        break
    except ValueError:
        # 找不到元素时，跳过当前行，继续下一次循环
        continue

print(f"行号：{row_idx}，列号：{col_idx}")

适配numpy ndarray的实现 #

如果你实际使用的是numpy的ndarray，不用手动遍历，直接用np.where()就能一次性获取所有匹配元素的行号和列号：

import numpy as np

arr = np.array([[0,1], [2,3]])
item = 2
# 获取所有匹配元素的行、列索引数组
rows, cols = np.where(arr == item)

if len(rows) > 0:
    # 取第一个匹配元素的索引
    print(f"行号：{rows[0]}，列号：{cols[0]}")
else:
    print("未找到目标元素")

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