python取索引为特定值的行dataframe
时间: 2024-01-24 12:02:03 浏览: 19
可以使用 `loc` 方法来获取索引为特定值的行。
例如,假设有以下的 DataFrame:
```
import pandas as pd
data = {
'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 30, 35, 40],
'gender': ['F', 'M', 'M', 'M']
}
df = pd.DataFrame(data)
df = df.set_index('name')
print(df)
```
输出:
```
age gender
name
Alice 25 F
Bob 30 M
Charlie 35 M
David 40 M
```
要获取索引为 "Charlie" 的行,可以使用以下代码:
```
row = df.loc['Charlie']
print(row)
```
输出:
```
age 35
gender M
Name: Charlie, dtype: object
```
如果要获取多个索引为特定值的行,可以传递一个列表给 `loc` 方法,例如:
```
rows = df.loc[['Alice', 'David']]
print(rows)
```
输出:
```
age gender
name
Alice 25 F
David 40 M
```
还可以使用 `iloc` 方法来根据整数索引获取行。例如,要获取第二行,可以使用以下代码:
```
row = df.iloc[1]
print(row)
```
输出:
```
age 30
gender M
Name: Bob, dtype: object
```