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

Làm cách nào để sửa đổi Scatterplot 2d để hiển thị màu dựa trên mảng thứ ba trong tệp CSV?

Để sửa đổi biểu đồ phân tán 2d để hiển thị màu dựa trên mảng thứ ba trong tệp CSV, chúng ta có thể thực hiện các bước sau -

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • Đọc tệp CSV có ba tiêu đề.
  • Tạo một hình mới hoặc kích hoạt một hình hiện có.
  • Thêm 'cái rìu' vào hình như một phần của sắp xếp ô phụ.
  • Tạo biểu đồ phân tán với các điểm dữ liệu tệp CSV.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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, df.data3, marker="*")

plt.show()

Tệp "input.csv" chứa dữ liệu sau

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

Đầu ra

Khi chúng tôi chạy mã, nó sẽ tạo ra biểu đồ phân tán sau

Làm cách nào để sửa đổi Scatterplot 2d để hiển thị màu dựa trên mảng thứ ba trong tệp CSV? Làm cách nào để sửa đổi Scatterplot 2d để hiển thị màu dựa trên mảng thứ ba trong tệp CSV?