python中pandas读取数据时保留特定名称的列
时间: 2023-11-09 17:04:16 浏览: 203
使用pandas读取csv文件的指定列方法
可以使用pandas中的loc方法来选择特定名称的列。例如,如果你的数据文件包含名为"Name"、"Age"、"Gender"和"Salary"的列,你可以使用以下代码来选择"Name"和"Salary"列:
```python
import pandas as pd
# 读取数据文件
data = pd.read_csv("data.csv")
# 选择特定列
selected_columns = data.loc[:, ["Name", "Salary"]]
# 打印选择的列
print(selected_columns)
```
在上面的代码中,`loc`方法用于选择所有行和特定列。`["Name", "Salary"]`是一个包含所需列名称的列表。你可以将其替换为你需要选择的列名称列表。
阅读全文