pandas赛选数据后选第一个
时间: 2023-10-13 13:27:19 浏览: 41
您可以使用 pandas 的 loc 函数或 iloc 函数来选择需要的数据行,然后使用 .iloc[0] 或 .loc[0] 来获取第一行的数据。例如:
使用 loc 函数:
```
import pandas as pd
df = pd.DataFrame({
'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]
})
filtered_df = df.loc[(df['A'] == 'foo') & (df['B'] == 'one')]
first_row = filtered_df.loc[filtered_df.index[0]]
print(first_row)
```
输出结果:
```
A foo
B one
C 1
D 10
Name: 0, dtype: object
```
使用 iloc 函数:
```
import pandas as pd
df = pd.DataFrame({
'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]
})
filtered_df = df.loc[(df['A'] == 'foo') & (df['B'] == 'one')]
first_row = filtered_df.iloc[0]
print(first_row)
```
输出结果:
```
A foo
B one
C 1
D 10
Name: 0, dtype: object
```
阅读全文