用 Python 编写一个程序,从 DataFrame 中打印"A"级学生的姓名

pythonpandasserver side programmingprogramming

输入

假设,您有 DataFrame,
 Id  Name Grade
0 1 stud1   A
1 2 stud2   B
2 3 stud3   C
3 4 stud4   A
4 5 stud5   A

输出

并且得到"A"级学生姓名的结果,

0    stud1
3    stud4
4    stud5

解决方案

为了解决这个问题,我们将遵循以下方法。

  • 定义一个 DataFrame

  • 将值与 DataFrame 进行比较

df[df['Grade']=='A']
  • 将结果存储在另一个 DataFrame 中并获取名称。

示例

让我们看看以下实现以获得更好的理解。

import pandas as pd
data =
[[1,'stud1','A'],[2,'stud2','B'],[3,'stud3','C'],[4,'stud4','A'],[5,'stud5','A']]
df = pd.DataFrame(data,columns=('Id','Name','Grade'))
print("DataFrame is\n",df)
print("find the A grade students name\n")
result = df[df['Grade']=='A']
print(result['Name'])

输出

DataFrame is
 Id Name Grade
0 1 stud1  A
1 2 stud2  B
2 3 stud3  C
3 4 stud4  A
4 5 stud5  A
find the A grade students name
0    stud1
3    stud4
4    stud5
Name: Name, dtype: object

相关文章