选择dataframe数据中的特定的列
时间: 2023-09-02 08:09:56 浏览: 94
您可以使用 Pandas 中的 `DataFrame` 类的 `loc` 或 `iloc` 方法来选择特定列。以下是一些示例代码:
1. 使用列名选择列:
```
import pandas as pd
# 创建一个示例 DataFrame
df = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35], 'City': ['New York', 'London', 'Paris']})
# 使用列名选择 'Name' 和 'Age' 列
selected_columns = df.loc[:, ['Name', 'Age']]
# 显示选择的列
print(selected_columns)
```
此代码将选择 'Name' 和 'Age' 列,并将结果存储在 `selected_columns` 变量中。
2. 使用列的索引选择列:
```
import pandas as pd
# 创建一个示例 DataFrame
df = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35], 'City': ['New York', 'London', 'Paris']})
# 使用列索引选择第一列和第三列
selected_columns = df.iloc[:, [0, 2]]
# 显示选择的列
print(selected_columns)
```
此代码将选择第一列和第三列,并将结果存储在 `selected_columns` 变量中。
阅读全文