📜  python 获取二维列表中项目的索引 - Python (1)

📅  最后修改于: 2023-12-03 14:46:17.963000             🧑  作者: Mango

Python 获取二维列表中项目的索引

在Python中,我们可以使用嵌套列表构造二维列表。如果需要获取二维列表中某个项目的索引,可以使用Python内置的函数 index()。下面是一个获取二维列表中项目索引的示例代码:

# 二维列表
my_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 获取元素 5 的索引
index_row = None
index_col = None

for i in range(len(my_list)):
    if 5 in my_list[i]:
        index_row = i
        index_col = my_list[i].index(5)
        break

print("元素 5 的索引为:({}, {})".format(index_row, index_col))

运行以上代码,将输出:元素 5 的索引为:(1, 1)。即元素 5 在二维列表中的索引为第2行第2列。

在上述示例代码中,我们使用了循环遍历二维列表,并使用内置函数 index() 获取元素的索引。需要注意的是,如果二维列表中存在重复元素,index() 仅会返回第一个匹配的索引。

另外,为了让代码更加简洁高效,我们可以将上述代码封装为一个函数,以便在后续的开发中进行调用。下面是一个封装后的函数示例:

def get_element_index(my_list, element):
    index_row = None
    index_col = None

    for i in range(len(my_list)):
        if element in my_list[i]:
            index_row = i
            index_col = my_list[i].index(element)
            break

    return (index_row, index_col)

在上述示例代码中,我们将获取元素索引的代码封装为了一个名为 get_element_index() 的函数,该函数接受两个参数,分别为需要进行查找的二维列表和目标元素。该函数会返回目标元素在二维列表中的索引。

使用该函数的示例如下:

# 二维列表
my_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 获取元素 5 的索引
index_row, index_col = get_element_index(my_list, 5)

print("元素 5 的索引为:({}, {})".format(index_row, index_col))

运行以上代码,将输出:元素 5 的索引为:(1, 1)。即元素 5 在二维列表中的索引为第2行第2列。

以上是关于 Python 获取二维列表中项目索引的介绍,推荐使用封装后的函数来进行操作,以提高代码的可重复性和效率。