Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python 팬더의 데이터 프레임에서 matplotlib 산점도 만들기

<시간/>

Pandas를 사용하여 데이터 프레임을 생성하고 subplot() 메서드를 사용하여 그림 및 축 변수를 생성할 수 있습니다. 그런 다음 ax.scatter() 메서드를 사용하여 필요한 플롯을 얻을 수 있습니다.

단계

  • 학생 수를 나열하십시오.

  • 학생들이 획득한 점수 목록을 작성하십시오.

  • 각 산점의 색상을 나타내기 위해 색상 목록을 가질 수 있습니다.

  • Pandas를 사용하여 데이터 프레임의 축을 나타내는 목록을 가질 수 있습니다.

  • 기본 nrows 및 ncols가 1인 subplots 방법을 사용하여 fig 및 ax 변수를 생성합니다.

  • plt.xlabel() 메서드를 사용하여 "학생 수" 레이블을 설정합니다.

  • plt.ylabel() 메서드를 사용하여 "획득한 마크" 레이블을 설정합니다.

  • 스캐터 포인트를 생성하려면 4단계에서 생성된 데이터 프레임을 사용하십시오. 포인트는 Students_count, 마크 및 색상입니다.

  • 그림을 보려면 plt.show() 메서드를 사용하십시오.

예시

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()

출력

Python 팬더의 데이터 프레임에서 matplotlib 산점도 만들기