原生python中没有数组的,只要列表
想要将数组转化为列表就用numpy.array(<list>)
方法
# coding=utf-8
# 二维列表中,每行从左到右递增,每列从上到下递增,给出一个数,判断它是否在列表中
# 从左下角(或右上角)开始遍历数组
def find_integer(matrix, num):
"""
:param matrix: [[]]
:param num: int
:return: bool
"""
if not matrix:
return False
rows, cols = len(matrix), len(matrix[0])
row, col = rows - 1, 0
while row >= 0 and col <= cols - 1:
if matrix[row][col] == num:
return True
elif matrix[row][col] > num:
row -= 1
else:
col += 1
return False
# 方案2 将列表转换为数组 用in 就行
import numpy as np
def find_integer_with_numpy(matrix, num):
if not matrix:
return False
matrix = np.array(matrix)
return num in matrix
if __name__ == '__main__':
matrix = [[1, 2, 3], [2, 3, 6], [3, 6, 7]]
num = 6
print(type(matrix)) # list
print(find_integer(matrix, num))
print(find_integer_with_numpy(matrix, num))