从一个csv文件选取特定的列
时间: 2023-11-24 10:07:00 浏览: 78
使用pandas读取csv文件的指定列方法
可以使用Python中的pandas库来读取CSV文件,并且选取特定的列。以下是一个示例代码:
```python
import pandas as pd
# 读取CSV文件
df = pd.read_csv('data.csv')
# 选取特定的列
selected_columns = ['column1', 'column2', 'column3']
df = df[selected_columns]
# 显示选取的列
print(df.head())
```
在这个示例代码中,我们首先使用`pd.read_csv()`函数来读取CSV文件。然后,我们使用一个列表来选取我们想要的列,并将结果保存回原始的DataFrame中。最后,我们使用`df.head()`来显示选取的列的前几行数据。
阅读全文