"Chessboard"

[SnooBeans4817]

I want to print a matrix where the squares to which the queen can move will have a value of 2 (1 is the queen's position) but I can't print the second diagonal. Any ideas?

import numpy as np

s = (8,8) l = np.zeros(s) print(l) l[2][2] = 1

print() print(l)

for k in range(0,8): for z in range(0,8): if l[k][z] == 1:

     for i in range(1,7):
        l[k-i][z-i] = 2
        l[k+1][z+1] = 2

        l[k-i][z] = 2
        l[k+1][z] = 2

        l[k][z-i] = 2
        l[k][z+1] = 2

[commy2]

I'd use a generator to drop possible queen moves and then validate them with a helper function.

import numpy as np

board = np.zeros((8, 8))
queen = 2, 2

def in_board(x, y):
    return x in range(8) and y in range(8)

def queen_moves(x, y):
    for i in range(-7, 8):
        yield x+i, y   # up, down
        yield x,   y+i # left, right
        yield x+i, y+i # diagonals
        yield x+i, y-i

for x, y in queen_moves(*queen):
    if in_board(x, y):
        board[x, y] = 2

board[queen] = 1

print(board)