python中寻找矩阵位置的函数find

在Python中，没有内置的矩阵查找函数find。不过可以使用numpy库来实现矩阵查找。

例如，假设有一个矩阵A：

import numpy as np

A = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

要查找元素5所在的位置，可以使用以下代码：

indices = np.where(A == 5)
print(indices)

输出结果为：

(array([1]), array([1]))

这表示元素5在矩阵A的第2行第2列。其中，np.where函数返回一个元组，包含元素5在行和列的索引。如果矩阵中有多个值相同的元素，np.where函数会返回它们的所有位置。

相关问题

用python，寻找矩阵中第i行j列的元素为矩阵中的最大元素

下面是一种可能的实现方式：

def find_max_in_matrix(matrix, i, j):
    """
    Find the maximum element in the i-th row and j-th column of the given matrix.
    """
    row_max = max(matrix[i])  # find the maximum element in the i-th row
    col_max = max(matrix[r][j] for r in range(len(matrix)))  # find the maximum element in the j-th column
    return max(row_max, col_max)  # return the overall maximum element

使用示例：

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

print(find_max_in_matrix(matrix, 1, 1))  # output: 9

这个函数首先通过 `max(matrix[i])` 找到了矩阵中第 `i` 行的最大元素，然后通过生成器表达式 `max(matrix[r][j] for r in range(len(matrix)))` 找到了矩阵中第 `j` 列的最大元素。最后，将这两个值取最大值就是整个矩阵中的最大元素。

在python中，编写一个函数，接收一个矩阵，返回此矩阵中的马鞍点。所谓马鞍点，即对于矩阵mxn 中的某个元素aij是第i行的最大值，同时又是第j列中的最小值，则称此元素为该矩阵中的一个马鞍点。

在Python中，你可以编写一个函数来找到矩阵中的马鞍点，可以按照以下步骤实现：

def find_saddle_point(matrix):
    row_max = [float('-inf')] * len(matrix)  # 初始化每行最大值为负无穷大
    col_min = [float('inf')] * len(matrix[0])  # 初始化每列最小值为正无穷大

    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if matrix[i][j] > row_max[i]:
                row_max[i] = matrix[i][j]
            if matrix[i][j] < col_min[j]:
                col_min[j] = matrix[i][j]

    saddle_points = []  # 存储马鞍点的列表
    for i, max_row in enumerate(row_max):
        for j, min_col in enumerate(col_min):
            if max_row == matrix[i][j] and min_col == matrix[i][j]:  # 检查当前元素是否既是行最大值也是列最小值
                saddle_points.append((i, j))

    return saddle_points if saddle_points else None  # 返回马鞍点的坐标列表，如果没有则返回None

# 示例
matrix = [
    [3, 7, 4],
    [5, 8, 6],
    [9, 1, 2]
]
saddle_points = find_saddle_point(matrix)
print("矩阵中的马鞍点是:", saddle_points)