如何修改二维散点图以基于 CSV 文件中的第三个数组显示颜色?

要修改二维散点图以根据 CSV 文件中的第三个数组显示颜色,我们可以采取以下步骤 -

  • 设置图形大小并调整子图之间和周围的填充。

  • 读取包含三个标题的 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数据2数据3
14571
29865
37529
45463
52312
63527
74639
85744
96851

输出结果

当我们运行代码时,它将产生以下散点图