Python用matplotlib库制作散点图

Python用matplotlib库制作散点图,有三列数据,如下所示,前两列数据分别作X轴、Y轴,剩余一列数据通过点的颜色渐变来表示不同的数据,在一开始,我将第三列数据(以列表形式)传递给scatte()的形参c引发报错,所以我也不知道该怎样办了。

img


这是我现在的代码:

from matplotlib.font_manager import FontProperties
import matplotlib.pyplot as plt
import matplotlib.pyplot as mp
import csv
import Functions


font = FontProperties(fname=r"c:\\windows\\fonts\\simsun.ttc")

plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']


filename = '..\\节气与舌象.csv'

x_v = []
y_v = []
color_v = []
with open(filename, encoding='gbk') as file:
    reader = csv.reader(file)

    for row in reader:
        x_v.append(row[0])
        y_v.append(row[1])
        color_v.append(row[2])
color = Functions.transform_list_to_numpy(color_v)
# mp.style.use('seaborn') 发生警告,并且错误显示
fig, ax = mp.subplots()
ax.scatter(x_v, y_v, c=color, cmap=mp.cm.Blues, s=10)  # c=color, cmap=mp.cm.Blues
ax.set_title('节气与舌象', fontsize=15, fontproperties=font)
ax.set_ylabel('舌象', fontsize=1, fontproperties=font)
ax.set_xlabel('节气', fontsize=10, fontproperties=font)
ax.tick_params(axis='both', which='major', labelsize=5)
mp.show()
# mp.savefig('节气与舌象.png', bbox_inches='tight')'
x_v = []
y_v = []
color_v = []
with open(filename, encoding='gbk') as file:
    reader = csv.reader(file)

    for row in reader:
        x_v.append(row[0])
        y_v.append(row[1])
        color_v.append(row[2])
color = Functions.transform_list_to_numpy(color_v)
# mp.style.use('seaborn') 发生警告,并且错误显示
fig, ax = mp.subplots()
ax.scatter(x_v, y_v, c=color, cmap=mp.cm.Blues, s=10)  # c=color, cmap=mp.cm.Blues
ax.set_title('节气与舌象', fontsize=15, fontproperties=font)
ax.set_ylabel('舌象', fontsize=1, fontproperties=font)
ax.set_xlabel('节气', fontsize=10, fontproperties=font)
ax.tick_params(axis='both', which='major', labelsize=5)
mp.show()
# mp.savefig('节气与舌象.png', bbox_inches='tight')

相关文件:Functions.py

import numpy as np


def transform_list_to_numpy(list_val: list):
    matrix = np.ndarray(list_val)
    return matrix

注:图片中的文件只是数据样式示例
请各位指点!

不知道你这个问题是否已经解决, 如果还没有解决的话:
  • 以下回答来自chatgpt:

    根据参考资料和代码,可知使用scatter()函数绘制散点图时,c参数需要传递一个表示颜色的数组。在你的代码中使用的第三列数据colors是一个代表颜色渐变的数组,但是scatter()函数的c参数只接受表示颜色的字符串或数值,而不支持渐变数组。因此,需要对第三列数据进行处理,将其转换为表示颜色的字符串或数值。

    一种解决方案是将颜色数据映射为固定的字符串或数值,可以使用线性映射的方法将colors中的数值映射到固定的范围内,再将映射后的结果传递给scatter()函数的c参数。下面是具体的解决方案和代码示例:

    1. 导入所需的库:
    import matplotlib.pyplot as plt
    import numpy as np
    
    1. 定义数据:
    x = [1, 2, 3, 4, 5]
    y = [10, 20, 30, 40, 50]
    colors = [0.1, 0.3, 0.5, 0.7, 0.9]  # 第三列数据,表示颜色渐变
    
    1. 将颜色数据线性映射到固定的范围内(例如0到1),使用numpy的interp()函数:
    color_range = np.linspace(0, 1, len(colors))  # 生成指定范围内的数值
    c_values = np.interp(colors, color_range, (0, 1))  # 将颜色数据映射到0到1的范围内
    
    1. 制作散点图并传递映射后的颜色数据给scatter()函数的c参数:
    plt.scatter(x, y, c=c_values)
    
    1. 显示图形:
    plt.show()
    

    通过将颜色数据映射为固定的字符串或数值,可以解决scatter()函数的c参数不能接受颜色渐变数组的问题。希望这个解决方案能帮助到你!如果还有其他问题,可以随时追问。


如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^