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

matplotlibpythondata visualization

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

  • 设置图形大小并调整子图之间和周围的填充。
  • 读取带有三个标题的 CSV 文件。
  • 创建新图形或激活现有图形。
  • 将"ax"添加到图形作为子图排列的一部分。
  • 使用 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"文件包含以下数据

data1data2data3
14571
29865
37529
45463
52312
63527
74639
85744
96851

输出

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


相关文章