为什么我的输出没有按预期返回？

Question

所以我写了这段代码

def diagsDownRight(M):
    n = len(M)
    m = [[''] * (n - i - 1) + row + [''] * i for i, row in enumerate(M)]
    return ([''.join(col) for col in zip(*m)]), [''.join(col[::-1]) for col in zip(*m)]
def diagsUpRight(M):
    n = len(M)
    m = [['']*i + row + ['']*(n-i-1) for i, row in enumerate(M)]
    return [''.join(col) for col in zip(*m)], [''.join(col[::-1]) for col in zip(*m)]

def rows(M):
    return ["".join(row) for row in M], ["".join(reversed(row)) for row in M]
def cols(M):
    return ["".join(col) for col in zip(*M)], [''.join(col[::-1]) for col in zip(*M)]
def contains_word(grid: list[list[str]], w: str):
    if w in diagsUpRight(grid):
        return w
    if w in diagsDownRight(grid):
        return w
    if w in rows(grid):
        return w
    if w in cols(grid):
        return w

print(contains_word(grid=[
["r","a","w","b","i","t"],
["x","a","y","z","c","h"],
["p","q","b","e","i","e"],
["t","r","s","b","o","g"],
["u","w","x","v","i","t"],
["n","m","r","w","o","t"]]
, w='raw'))

对于此代码，如果在 rows(M)、cols(M)、diagsDownRight(M)、diagsUpRight(grid) 中找到单词 w，我想让 contains_word 返回单词 w，但是当我输入代码时，如图所示上面，输出不显示。我在这里做错了什么？

编辑我尝试这样做，但输出仍然返回 None

def contains_word(grid: list[list[str]], w: str):
    for col in diagsUpRight(grid):
        if w in col:
            return w
    for col in diagsDownRight(grid):
        if w in col:
            return w
    for row in rows(grid):
        if w in row:
            return w
    for col in cols(grid):
        if w in col:
            return w

print(contains_word(grid=[["r","a","w","b","i","t"],
["x","a","y","z","c","h"],
["p","q","b","e","i","e"],
["t","r","s","b","o","g"],
["u","w","x","v","i","t"],
["n","m","r","w","o","t"]]
, w='raw'))

Answer 1

错误：-

• 在您的代码中，每个 function 返回 tuples 并带有 2 个列表。
• 您在第二个代码的列表中进行了迭代，但 返回的 字符串是 'rawbit'，因此您需要遍历 'rawbit' 以获取 'raw'。

代码：-

def diagsDownRight(M):
    n = len(M)
    m = [[''] * (n - i - 1) + row + [''] * i for i, row in enumerate(M)]
    return [''.join(col) for col in zip(*m)]+[''.join(col[::-1]) for col in zip(*m)]

def diagsUpRight(M):
    n = len(M)
    m = [['']*i + row + ['']*(n-i-1) for i, row in enumerate(M)]
    return [''.join(col) for col in zip(*m)]+[''.join(col[::-1]) for col in zip(*m)]

def rows(M):
    return ["".join(row) for row in M]+["".join(reversed(row)) for row in M]

def cols(M):
    return ["".join(col) for col in zip(*M)]+ [''.join(col[::-1]) for col in zip(*M)]

def contains_word(grid: list[list[str]], w: str):
    for words in diagsUpRight(grid):
        if w in words:
            return w
    for words in diagsDownRight(grid):
        if w in words:
            return w
    for words in rows(grid):
        if w in words:
            return w
    for words in cols(grid):
        if w in words:
            return w


print(contains_word(grid=[["r","a","w","b","i","t"],
["x","a","y","z","c","h"],
["p","q","b","e","i","e"],
["t","r","s","b","o","g"],
["u","w","x","v","i","t"],
["n","m","r","w","o","t"]]
, w='raw'))

Answer 2

它返回 None 然后你有一个列表列表， 所以你需要在 contains_word 函数中遍历网格， 目前你浏览了一个列表列表，这个列表并没有解决你的 if in 条件

Answer 3

您原来的rows(M) 函数返回一个元组(['rawbit', 'xayzch', 'pqbeie', 'trsbog', 'uwxvit', 'nmrwot'], ['tibwar', 'hczyax', 'eiebqp ',' gobsrt ',' tivxwu ',' towrmn '])。为了满足条件if w in rows(grid):，w 必须匹配元组的列表之一，例如列表['rawbit', 'xayzch', 'pqbeie', 'trsbog', 'uwxvit ',' nmrwot ']。但是由于您正在寻找字符串raw，因此永远不会满足此条件。

特别是这样改rows(M)函数，结果是raw

def rows(M):
    return ' '.join(''.join(l) for l in M) + ' ' + ' '.join(''.join(reversed(l)) for l in M) # rawbit xayzch pqbeie trsbog uwxvit nmrwot tibwar hczyax eiebqp gobsrt tivxwu towrmn

contains_word 返回：

raw