如何修改 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"文件包含以下数据
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 |
输出
当我们运行代码时,它将产生以下散点图