ndarray转dict
时间: 2023-11-14 07:04:38 浏览: 82
python3 dict ndarray 存成json,并保留原数据精度的实例
可以使用numpy库中的tolist()方法将ndarray转换为Python列表,再使用Python内置的字典推导式将列表转换为字典。示例代码如下:
```python
import numpy as np
# 创建一个ndarray
arr = np.array([[1, 2], [3, 4]])
# 将ndarray转换为Python列表
lst = arr.tolist()
# 将列表转换为字典
my_dict = {i: lst[i] for i in range(len(lst))}
print(my_dict)
```
输出结果为:
```
{0: [1, 2], 1: [3, 4]}
```
其中,字典推导式 ` {i: lst[i] for i in range(len(lst))}` 的作用是将列表 `lst` 转换为字典 `my_dict`,其中字典的键是列表下标,值是列表元素。
阅读全文