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

Tạo các biểu đồ phân tán matplotlib từ các khung dữ liệu trong gấu trúc của Python

Sử dụng Pandas, chúng ta có thể tạo khung dữ liệu và có thể tạo biến hình và trục bằng cách sử dụng phương thức subplot (). Sau đó, chúng ta có thể sử dụng phương thức ax.scatter () để lấy cốt truyện cần thiết.

Các bước

  • Lập danh sách số lượng học sinh.

  • Lập danh sách các điểm học sinh đã đạt được.

  • Để biểu thị màu của từng điểm phân tán, chúng ta có thể có một danh sách các màu.

  • Sử dụng Pandas, chúng ta có thể có một danh sách đại diện cho các trục của khung dữ liệu.

  • Tạo các biến fig và ax bằng phương pháp subplots, trong đó nrows và ncols mặc định là 1.

  • Đặt nhãn "Số học sinh" bằng phương thức plt.xlabel ().

  • Đặt nhãn "Điểm đạt được" bằng phương thức plt.ylabel ().

  • Để tạo điểm phân tán, hãy sử dụng khung dữ liệu đã tạo ở bước 4. Điểm là số lượng sinh viên, điểm và màu.

  • Để hiển thị hình này, hãy sử dụng phương thức plt.show ().

Ví dụ

from matplotlib import pyplot as plt
import pandas as pd

no_of_students = [1, 2, 3, 5, 7, 8, 9, 10, 30, 50]
marks_obtained_by_student = [100, 95, 91, 90, 89, 76, 55, 10, 3, 19]
color_coding = ['red', 'blue', 'yellow', 'green', 'red', 'blue', 'yellow', 'green', 'yellow', 'green']

df = pd.DataFrame(dict(students_count=no_of_students,
marks=marks_obtained_by_student, color=color_coding))

fig, ax = plt.subplots()

plt.xlabel('Students count')
plt.ylabel('Obtained marks')

ax.scatter(df['students_count'], df['marks'], c=df['color'])

plt.show()

Đầu ra

Tạo các biểu đồ phân tán matplotlib từ các khung dữ liệu trong gấu trúc của Python