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

CSV 파일의 세 번째 배열을 기반으로 색상을 표시하도록 2D 산점도를 수정하는 방법은 무엇입니까?

<시간/>

CSV 파일의 세 번째 배열을 기반으로 색상을 표시하도록 2D 산점도를 수정하려면 다음 단계를 수행할 수 있습니다. -

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • 3개의 헤더가 있는 CSV 파일을 읽습니다.
  • 새 그림을 만들거나 기존 그림을 활성화합니다.
  • 하위 플롯 배열의 일부로 그림에 '도끼'를 추가합니다.
  • CSV 파일 데이터 포인트로 산점도를 만듭니다.
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import pandas as pd
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

columns = ["data1", "data2", "data3"]
df = pd.read_csv("input.csv", usecols=columns)

fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(df.data1, df.data2, df.data3, marker="*")

plt.show()

"input.csv" 파일에는 다음 데이터가 포함되어 있습니다.

데이터1 data2 data3
1 45 71
2 98 65
3 75 29
4 54 63
5 23 12
6 35 27
7 46 39
8 57 44
9 68 51

출력

코드를 실행하면 다음과 같은 산점도가 생성됩니다.

CSV 파일의 세 번째 배열을 기반으로 색상을 표시하도록 2D 산점도를 수정하는 방법은 무엇입니까? CSV 파일의 세 번째 배열을 기반으로 색상을 표시하도록 2D 산점도를 수정하는 방법은 무엇입니까?