PyCharm 在以下代码中抛出黄色警告，它似乎认为函数的输出是ndarray[Any, dtype]。我应该忽略这个错误吗？或者有没有 Pythonic 方法来处理这个问题？

我也尝试过np.array(articles, dtype=str)，np.array(articles, dtype=np.str_)但都无法解决问题。

import numpy as np


def load_data() -> np.ndarray[str]:
    articles = []
    with open("data/bbc_raw/001.txt", "rt") as f:
        articles.append(f.read().strip())
    return np.array(articles)


if __name__ == "__main__":
    array = load_data()

警告：

Expected type 'ndarray[str]', got 'ndarray[Any, dtype]' instead

解决方案

因此，正如其他人所指出的那样，np.typing.NDArray这确实是可行的方法，但并不是那么简单。为了指定数组的类型，Numpy 要求您使用可以出现在数组中的每种类型的特殊版本，例如：

• str->np.str_
• int->np.int_
• float->np.float_

就我个人而言，我认为上述语法并不完全理想，但它似乎是它预期的工作方式。

因此，为了指定没有类型错误的上述代码，您可以像这样创建返回类型：

import numpy as np
from numpy.typing import NDArray

def load_data() -> NDArray[np.str_]:
    articles = []
    with open("data/bbc_raw/001.txt", "rt") as f:
        articles.append(f.read().strip())
    return np.array(articles)


if __name__ == "__main__":
    array = load_data()

希望这有用！