怎么将DataFrame中的某几列重新生成一个DataFrame
时间: 2023-07-25 22:09:43 浏览: 103
将一个DataFrame中的一列(行),插入到另一个DataFrame中
可以使用`pandas`库的`DataFrame`中的`loc`函数,选择需要的列并生成新的`DataFrame`。
比如,假设有以下`DataFrame`:
```
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]})
```
如需将'A'和'B'列重新生成一个新的`DataFrame`,可以使用以下代码:
```
new_df = df.loc[:, ['A', 'B']]
```
其中`loc`函数中的第一个参数为行索引的范围,使用':'表示选择所有行;第二个参数为需要选择的列名,使用列表格式表示。以上代码将生成以下新的`DataFrame`:
```
A B
0 1 4
1 2 5
2 3 6
```
阅读全文