def get_column(square_size, square, i):
    """Returns an i-th column of a square."""
    expected_modulo = i % square_size
    col = []
    for i, item in enumerate(square):
        if i % square_size == expected_modulo:
            col.append(item)
    return col