python把不同维度的数组写入数据帧
时间: 2023-10-07 18:12:18 浏览: 85
可以使用 pandas 库将不同维度的数组转换为 Dataframe。以下是示例代码:
```python
import pandas as pd
import numpy as np
# 创建一个包含不同维度数组的字典
data = {'col1': np.array([1, 2, 3]),
'col2': np.array([[4, 5], [6, 7], [8, 9]]),
'col3': np.array([[10, 11, 12], [13, 14, 15]])}
# 将字典转换为 Dataframe
df = pd.DataFrame.from_dict(data, orient='index').T
# 打印 Dataframe
print(df)
```
输出:
```
col1 col2 col3
0 1 [4, 5.0] [10.0, 11.0, 12.0]
1 2 [6, 7.0] [13.0, 14.0, 15.0]
2 3 [8, 9.0] NaN
```
在上面的代码中,我们首先创建一个包含不同维度数组的字典。然后,我们使用 `pd.DataFrame.from_dict()` 函数将其转换为 Dataframe。我们需要将 `orient` 参数设置为 `'index'`,以便正确处理字典中的不同维度数组。最后,我们使用 `.T` 将 Dataframe 翻转,以便正确显示数组的维度。在这个例子中,我们创建了一个包含三个列的 Dataframe,每列包含不同维度的数组。你可以根据你自己的需求修改字典中的数组和维度。
阅读全文