矩阵python函数

时间:2018-11-12 15:58:11

标签: python matrix

我必须编写一个函数,给定一个非空且不为负的矩阵,它是matrix[i][j]的简写形式的起始位置(代表矩阵位置的长度为2的非负列表),返回最低相邻值的位置(保持吉普音直到找到最低的值才通过相邻的单元格/元素)。

示例:如果matrix = [[8,90,91,73],[60,6,32,84],[50,4,45,94],[12,85,3,2]]且销售位置= [0,0],则应返回[3,3]。我想尝试仅使用编程的基础知识(如果语句,循环等)来实现此功能,可以使用诸如min或max之类的简单辅助函数,但是请不要太高级。

这是我到目前为止所拥有的,但是由于索引超出范围或范围错误,它并不能很好地工作:

def search_local_lowest_value(s: List[List[int]], position: List[int]) -> List[int]:
element1 = position[0]
element2 = position[1]
local_variable = s[element1][element2]

while local_variable <= s[row + 1][column] or local_variable <= s[row + 1][column + 1] \
    or local_variable <= s[row + 1][column - 1] or local_variable <= s[row][column + 1] \
    or local_variable <= s[row][column - 1] or local_variable <= s[row -1][column] \
    or local_variable <= s[row - 1][column + 1] or local_varaible <= s[row - 1][column - 1]:
        if local variable <= s[row + 1][column] or local_variable <= s[row + 1][column + 1] \
        or local_variable <= s[row + 1][column - 1] or local_variable <= s[row][column + 1] \
        or local_variable <= s[row][column - 1] or local_variable <= s[row -1][column] \
        or local_variable <= s[row - 1][column + 1] or local_variable <= s[row - 1][column - 1]:
            return min() // don't know what to fill for that
        else: 
            return False

1 个答案:

答案 0 :(得分:0)

我想出了这个解决方案,我使用了更加清晰干燥的方法:

def bordering(start, shape):
  PATTERN = [[-1,-1], [-1, 0], [-1, 1], [0, 0], [0, -1], [0, 1], [1, -1], [1, 0], [1, 1]]
  border = [[start[0] + x, start[1] + y] for x, y in PATTERN]
  valid_border = [item for item in border if item[0] >= 0 and item[1] >= 0 and item[0] <= shape[0] - 1   and item[1] <= shape[1] - 1  ] # removes borders out of the matrix boundaries
  return valid_border

def smaller_from(start, matrix):
  shape = [len(matrix), len(matrix[0])]
  borders = bordering(start, shape)
  minimum = min([ [matrix[x][y], [x,y]] for x,y in borders ]) # maps the values and coordinates in valid borders and returns the minimum
  return minimum

def find_minimum(start, matrix):
  result = []
  while True:
    val_coords = smaller_from(start, matrix)
    result = val_coords
    if val_coords[0] >= matrix[start[0]][start[1]]:
      break
    else:
      start = val_coords[1]
  return result

matrix = [
          [8,90,91,73],
          [60,6,32,84],
          [50,4,45,94],
          [12,85,3,2]]
start = [0, 0]

find_minimum(start, matrix) #=> [2, [3, 3]]