comefrom: <doocs>1424.Diagonal Traverse II_1.py
code: def findDiagonalOrder( nums: List[List[int]]) -> List[int]:
      arr = []
      for i, row in enumerate(nums):
          for j, v in enumerate(row):
              arr.append((i + j, j, v))
      arr.sort()
      return [v[2] for v in arr]
masked: def sample_funcname ( var_0 : List [ List [ int ] ] ) -> List [ int ] :
      var_3 = [ ]
      for var_5 , var_6 in enumerate ( var_0 ) :
          for var_7 , var_8 in enumerate ( var_6 ) :
              var_3.append ( ( var_5 + var_7 , var_7 , var_8 ) )
      var_3.sort ( )
      return [ var_8 [ 2 ] for var_8 in var_3 ]
  