pandas read_csv指定列
时间: 2023-08-21 14:08:18 浏览: 91
pandas中read_csv的缺失值处理方式
您可以使用pandas的read_csv方法来读取CSV文件,并通过指定列参数来选择需要读取的列。
下面是一个示例代码:
```python
import pandas as pd
# 读取整个CSV文件
data = pd.read_csv('data.csv')
# 选择需要读取的列
selected_columns = ['column1', 'column2'] # 请将'column1'和'column2'替换为实际需要选择的列名
data_selected = data[selected_columns]
# 打印选择的列数据
print(data_selected)
```
在上面的示例代码中,`data.csv`是要读取的CSV文件名,`selected_columns`是一个包含需要选择的列名的列表。read_csv方法会读取整个CSV文件,并将所选列的数据存储在`data_selected`变量中。
请确保将'column1'和'column2'替换为实际需要选择的列名。您还可以根据需要选择更多的列,并将它们添加到`selected_columns`列表中。
阅读全文