如何修改2D散点图以根据CSV文件中的第三个数组显示颜色？

要修改2D散点图以根据CSV文件中的第三个数组显示颜色，可以执行以下步骤−

• 设置图形大小并调整子图之间和周围的填充。
• 读取具有三个标题的CSV文件。
• 创建新的图形或激活现有图形。
• 将“ax”作为子图排列的一部分添加到图形中。
• 使用CSV文件数据点绘制散点图。
• 要显示图形，请使用 show() 方法。

阅读更多：Python 教程

示例

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”文件包含以下数据

输出

运行该代码，将生成以下散点图