dataframe选取指定列
时间: 2023-11-18 14:06:35 浏览: 87
使用pandas库的DataFrame可以轻松地选择指定列。以下是一些示例:
假设我们有一个DataFrame df,其中包含以下列:'A'、'B'、'C'、'D'、'E'。
1. 选取单列
要选择单列,可以使用以下语法:
```python
df['A']
```
这将返回一个Series,其中包含'A'列的所有行。
2. 选取多列
要选择多列,可以将列名作为列表传递给DataFrame:
```python
df[['A', 'B', 'C']]
```
这将返回一个DataFrame,其中包含'A'、'B'和'C'列的所有行。
3. 切片列
要选择列的子集,可以使用切片操作:
```python
df.loc[:, 'A':'C']
```
这将返回一个DataFrame,其中包含'A'、'B'和'C'列的所有行。
4. 布尔索引
要根据条件选择列,可以使用布尔索引:
```python
df[df['A'] > 0]
```
这将返回一个DataFrame,其中包含所有'A'列大于0的行。
5. 使用iloc
使用iloc可以按位置选择列。以下示例选择第1列、第3列和最后一列:
```python
df.iloc[:, [0, 2, -1]]
```
阅读全文