将数据框中的两列提出生成一个新的数据集
时间: 2024-02-15 14:05:49 浏览: 47
假设要将数据框`df`中的两列`col1`和`col2`提出生成一个新的数据集,可以使用`loc`方法或者直接用列名索引的方式进行操作。
方法1:使用`loc`方法
```python
new_df = df.loc[:, ['col1', 'col2']]
```
方法2:使用列名索引的方式
```python
new_df = df[['col1', 'col2']]
```
两种方法的效果是一样的,都可以提取出`col1`和`col2`列,生成一个新的数据集`new_df`。需要注意的是,这种方法只是提取了`col1`和`col2`列,如果需要保留其他列,可以在`loc`方法或者列名索引的方式中加入需要保留的列名。例如:
```python
new_df = df.loc[:, ['col1', 'col2', 'col3']]
```
或者:
```python
new_df = df[['col1', 'col2', 'col3']]
```
其中`col3`是需要保留的列名。
阅读全文