如何将dataframe按列分为两行,例如:将1,2,3,4,5分为第一行(1,2,3)和第二行(1,4,5)
时间: 2024-03-05 16:51:48 浏览: 25
Python DataFrame一列拆成多列以及一行拆成多行
5星 · 资源好评率100%
可以按照以下步骤将dataframe按列分为两行:
1. 创建一个包含所有列的列表,并将列分为两个子列表,分别代表两行。
```
all_cols = list(df.columns)
row1_cols = all_cols[:3]
row2_cols = all_cols[::2]
```
2. 使用子列表创建两个新的dataframe。
```
df_row1 = df[row1_cols]
df_row2 = df[row2_cols]
```
其中,`df[row1_cols]`表示只选择前三列,`df[row2_cols]`表示每隔一列选择一次。
完整代码如下:
```
import pandas as pd
# 创建一个包含所有列的dataframe
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15],
'D': [16, 17, 18, 19, 20],
'E': [21, 22, 23, 24, 25]})
# 将列分为两个子列表
all_cols = list(df.columns)
row1_cols = all_cols[:3]
row2_cols = all_cols[::2]
# 创建两个新的dataframe
df_row1 = df[row1_cols]
df_row2 = df[row2_cols]
# 打印结果
print("第一行:\n", df_row1)
print("第二行:\n", df_row2)
```
输出结果为:
```
第一行:
A B C
0 1 6 11
1 2 7 12
2 3 8 13
3 4 9 14
4 5 10 15
第二行:
A C E
0 1 11 21
1 2 12 22
2 3 13 23
3 4 14 24
4 5 15 25
```
阅读全文