如何将特定类型的 numpy 数组声明为数据类中的类型

Ani*_*nia 6 python-3.x numpy-ndarray python-dataclasses

我所拥有的:
我正在创建一个数据类,并说明其元素的类型:

class Task():  
     n_items: int  
     max_weight: int  
     max_size: int  
     items: numpy.array(Item)  # incorrect way of doing it
Run Code Online (Sandbox Code Playgroud)

我想做的事情是
我想声明,该items将是“Item”类的对象的 numpy 数组

Grz*_*ski 5

你可以把ndarray

import numpy as np

class Task():  
     n_items: int  
     max_weight: int  
     max_size: int  
     items: np.ndarray
Run Code Online (Sandbox Code Playgroud)