pandas的loc函数
时间: 2023-10-16 22:29:12 浏览: 34
pandas的`loc`函数是用于通过标签或布尔数组选择行和列的方法。
具体而言,`loc`函数可以通过以下方式选择行和列:
- 通过标签选择:可以通过指定行标签和列标签来选择特定的行和列。例如:`df.loc[row_labels, col_labels]`。
- 通过布尔数组选择:可以通过指定一个布尔数组来选择特定的行和列。例如:`df.loc[boolean_row_array, boolean_col_array]`。
此外,`loc`函数还可以选择所有行或列(使用冒号 `:` 符号)以及选择特定的行或列(使用列表或切片符号 `[]`)。
示例代码:
```python
import pandas as pd
# 创建一个示例DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'],
'age': [25, 30, 35, 40, 45],
'city': ['Beijing', 'Shanghai', 'Guangzhou', 'Shenzhen', 'Chengdu']}
df = pd.DataFrame(data)
# 通过标签选择
print(df.loc[1:3, 'name':'age'])
# 通过布尔数组选择
boolean_row_array = df['age'] > 30
boolean_col_array = ['name', 'city']
print(df.loc[boolean_row_array, boolean_col_array])
```
输出结果:
```
name age
1 Bob 30
2 Charlie 35
name city
2 Charlie Guangzhou
3 David Shenzhen
4 Emily Chengdu
```