numpy.ndarray转换为原始数据

### 将 NumPy `ndarray` 对象转换为原始 Python 数据类型当处理数值类型的 `ndarray` 时，可以使用 `.item()` 方法将其单个元素转换回 Python 的内置数据类型。对于整个数组，则可以通过调用 `.tolist()` 方法实现转换。 #### 单个元素转为 Python 原始类型 ```python import numpy as np arr = np.array([1]) element = arr.item() print(type(element)) # <class 'int'> ``` #### 整个数组转为 Python 列表 ```python import numpy as np arr = np.array([[1, 2], [3, 4]]) list_data = arr.tolist() print(list_data) # [[1, 2], [3, 4]] print(type(list_data)) # <class 'list'> ``` 对于结构化数组（即具有复合数据类型的数组），同样适用上述方法： ```python dtype_student = [('name', 'U10'), ('age', 'i4'), ('score', 'f4')] structured_array = np.array([('abc', 21, 50), ('xyz', 18, 75)], dtype=dtype_student) # 转换为列表形式 student_list = structured_array.tolist() for student in student_list: name, age, score = student print(f"Name: {name}, Age: {age}, Score: {score}") ``` 此外，在某些情况下可能需要将特定的数据类型强制转换为目标类型，这可通过 `astype()` 函数完成[^2]。

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇如何在transformers中的loss_utils中添加自己的损失函数