dataframe保留其中两列形成新的dataframe
时间: 2023-11-18 09:07:22 浏览: 95
对dataframe数据之间求补集的实例详解
假设你有一个名为 `df` 的 DataFrame,其中包含多列数据,你想保留其中两列形成新的 DataFrame,你可以使用 `loc` 或 `iloc` 方法进行切片操作。
例如,假设你想保留 `df` 中的 `column1` 和 `column2` 两列,你可以使用以下代码创建一个新的 DataFrame:
```python
new_df = df.loc[:, ['column1', 'column2']]
```
或者使用以下代码:
```python
new_df = df.iloc[:, [0, 1]]
```
上述代码中,`loc` 方法使用列名进行切片,而 `iloc` 方法使用列的索引进行切片。在这两种情况下,`:` 表示选取所有行。
阅读全文