python中pandas读取数据时保留某几列
时间: 2023-07-28 09:15:41 浏览: 167
Python读取txt某几列绘图的方法
在pandas中,可以使用`usecols`参数来选择需要保留的列。例如,如果我们有一个名为`data.csv`的数据集,其中有4列,我们只想保留其中的2列,可以使用以下代码:
```python
import pandas as pd
# 读取数据集,只保留第1列和第3列
data = pd.read_csv('data.csv', usecols=[0,2])
# 打印保留的列
print(data.columns)
```
这将只选择第1列和第3列,并将其存储在一个名为`data`的pandas DataFrame中。我们可以使用`data.columns`打印保留的列名。
阅读全文