将m*n维的numpy数组转化为字典,其中第一列为key,后几列为该key的值且为列表形式
时间: 2024-03-05 10:49:17 浏览: 68
python字典值排序并取出前n个key值的方法
可以使用numpy模块的ndarray类型的tolist()方法将数组转换为Python的内置列表类型,然后使用Python的字典推导式生成字典。具体代码如下:
```python
import numpy as np
# 创建一个m*n维的numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将数组转换为Python内置的列表类型
lst = arr.tolist()
# 使用字典推导式生成字典
my_dict = {row[0]: row[1:] for row in lst}
print(my_dict)
```
在上面的代码中,我们首先创建了一个3*3维的numpy数组,然后将其转换为Python内置的列表类型。接着,我们使用字典推导式生成字典,其中键为每一行的第一个元素,值为该行除第一个元素外的所有元素组成的列表。最后,打印生成的字典,输出如下:
```
{1: [2, 3], 4: [5, 6], 7: [8, 9]}
```
其中,字典的键对应原数组的第一列,后面的列表对应原数组的剩余列。
阅读全文