Computer >> Máy Tính >  >> Lập trình >> Python

Viết chương trình bằng Python để in tên "học sinh lớp A" từ DataFrame

Đầu vào -

Assume, you have DataFrame,
 Id  Name Grade
0 1 stud1   A
1 2 stud2   B
2 3 stud3   C
3 4 stud4   A
4 5 stud5   A

Đầu ra -

Và kết quả cho tên học sinh lớp ‘A’,

0    stud1
3    stud4
4    stud5

Giải pháp

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các cách tiếp cận dưới đây.

  • Xác định DataFrame

  • So sánh giá trị với DataFrame

df[df['Grade']=='A']
  • Lưu trữ kết quả trong DataFrame khác và tìm nạp Tên.

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn.

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'])

Đầu ra

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