将m*n维的numpy数组转化为dataframe,其中每一行第一个值为索引,后面为为numpy数组,利用pandas实现
时间: 2024-03-05 08:49:58 浏览: 78
可以使用 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)
# 重命名列名
df.columns = ['value1', 'value2', 'value3']
print(df)
```
输出结果为:
```
value1 value2 value3
0
1 2 3 4
2 4 6 8
3 6 9 12
```
其中,第一列作为索引列,后续列作为数值列。
相关问题
将m*n维的numpy数组转化为字典,其中每一行第一个值为索引,后面为为numpy数组,利用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 为一个列表,对应着该行的后面几列的数值。
将m*n维的numpy数组转化为字典,其中第一列为key,后几列为该key的值且为数组形式,如果用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, columns=['key', 'value1', 'value2', 'value3'])
# 将 DataFrame 转化为字典
result_dict = df.groupby('key').apply(lambda x: x.iloc[:,1:].values.tolist()).to_dict()
print(result_dict)
```
输出结果为:
```
{1: [[2, 3, 4]], 2: [[4, 6, 8]], 3: [[6, 9, 12]]}
```
其中,字典的 key 对应着第一列的数值,value 为一个二维列表,每个元素为一个数组,对应着该 key 的后面几列的数值。
阅读全文