dataframe选取某两列
时间: 2023-07-06 20:41:38 浏览: 94
可以使用`loc`或`iloc`方法选取某两列。这里给出两种方法的示例:
1. 使用`loc`方法选取某两列,例如选取`name`和`age`两列:
```python
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 32, 18, 47],
'city': ['New York', 'Paris', 'London', 'Tokyo']})
# 选取name和age两列
df_subset = df.loc[:, ['name', 'age']]
print(df_subset)
```
输出结果如下:
```
name age
0 Alice 25
1 Bob 32
2 Charlie 18
3 David 47
```
2. 使用`iloc`方法选取某两列,例如选取第一列和第二列:
```python
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 32, 18, 47],
'city': ['New York', 'Paris', 'London', 'Tokyo']})
# 选取第一列和第二列
df_subset = df.iloc[:, [0, 1]]
print(df_subset)
```
输出结果如下:
```
name age
0 Alice 25
1 Bob 32
2 Charlie 18
3 David 47
```
需要注意的是,使用`loc`方法选取列时,需要使用列名;而使用`iloc`方法选取列时,需要使用列的索引。
阅读全文