关于元素的python矩阵问题

我在这个问题上有一个矩阵([1 3 4],[4 5 6],[7 8 9])我必须找到两个元素之间的差,即同时从行和列中<= 2并通过当差异<= 2时自动在其他元素上

from numpy import *

arr1= array([1,4,3,4,5,6,7,8,9])

arr2= arr1.reshape(3,3)

print(arr2)

x=int(input("no.of test cases"))

i=0

j=0

for i in range(3):

    for j in range(3):

        y=arr2[i+1][j+1]-arr2[i+1][j]

        if(y<=2):

           i=i+1

           j=j+1

print(y)

这是我的代码,因为我是python的新手,所以我收到很多错误

评论
  • amodi
    amodi 回复

    Python列表和numpy数组的索引均为0,因此请勿在数组访问中添加1。

    从numpy import *

    arr1 =数组([1,4,3,4,5,6,7,8,9])

    arr2 = arr1.reshape(3,3)

    打印(arr2)

    x = int(input(“测试用例数”))

    i = 0

    j = 0

    对于我在范围(3)中:

    for j in range(3):
    
        y=arr2[i][j]-arr2[i][j]
    
        if(y<=2):
    
           i=i+1
    
           j=j+1
    

    打印(y)