Đọc file CSV có 3 cột (data1, data2, data3) bằng Pandas, vẽ scatter plot với x=data1, y=data2, màu (color) theo data3, marker dạng sao (*).
Ví dụ
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, c=df.data3, marker="*")
plt.show()
File "input.csv" dữ liệu mẫu:
| data1 | 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 |
Kết quả
Scatter plot các điểm (data1, data2) màu theo giá trị data3, marker sao.