将m*n维的numpy数组转化为字典,其中每一行第一个值为索引,后面为为numpy数组,利用pandas实现
时间: 2024-03-05 10:49:57 浏览: 67
使用pandas把某一列的字符值转换为数字的实例
可以使用 pandas 实现该操作,具体代码如下:
```python
import pandas as pd
import numpy as np
# 创建一个 3x4 的numpy数组
arr = np.array([[1,2,3,4], [2,4,6,8], [3,6,9,12]])
# 将数组转化为 DataFrame
df = pd.DataFrame(arr)
# 将第一列设置为索引列
df.set_index(0, inplace=True)
# 将 DataFrame 转化为字典
result_dict = df.apply(lambda x: x.values.tolist(), axis=1).to_dict()
print(result_dict)
```
输出结果为:
```
{1: [2, 3, 4], 2: [4, 6, 8], 3: [6, 9, 12]}
```
其中,字典的 key 对应着第一列的数值,value 为一个列表,对应着该行的后面几列的数值。
阅读全文