Python - ปรับ Scatter Plot 2D ให้สีตามคอลัมน์ที่ 3 จาก CSV

อ่าน CSV 3 คอลัมน์ (x, y, color) แล้ววาด Scatter Plot สีตามคอลัมน์ที่ 3

ตัวอย่างโค้ด

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

คำอธิบาย

• pd.read_csv(usecols=...) อ่านเฉพาะคอลัมน์ที่ต้องการ
• ax.scatter(x, y, c) พารามิเตอร์ที่ 3 c ใช้กำหนดสี (Color Mapping)
• marker="*" ใช้ Marker ดาว
• ค่า data3 จะถูก Map เป็นสีตาม Colormap Default (Viridis)