pandas 等分切分
时间: 2023-07-01 13:22:14 浏览: 122
python按照指定量级进行文件拆分 pandas
如果您想要将 Pandas 数据框按行等分成多个子数据框,可以使用 numpy 库中的 `array_split` 函数。首先,您需要计算要将数据框划分成多少个子数据框。接下来,使用 `array_split` 函数将数据框分成相应的子数据框。
以下是一个示例代码:
```python
import pandas as pd
import numpy as np
# 创建一个示例数据框
df = pd.DataFrame({'A': [1, 2, 3, 4, 5, 6], 'B': ['a', 'b', 'c', 'd', 'e', 'f']})
# 计算要划分成多少个子数据框
num_splits = 3
split_sizes = [len(df) // num_splits] * num_splits
split_sizes[-1] += len(df) % num_splits
# 使用array_split函数将数据框分成相应的子数据框
sub_dfs = np.array_split(df, num_splits)
print(sub_dfs)
```
在上面的代码中,我们首先计算了要将数据框划分成几个子数据框,然后计算了每个子数据框的大小。最后,我们使用 `array_split` 函数将数据框划分成相应的子数据框,并将结果存储在 `sub_dfs` 变量中。
阅读全文