如何在Python对数刻度图中用第三变量控制颜色渐变？

你的问题出在loglog方法的c参数不支持直接传入数组实现颜色映射，要达成你想要的效果，推荐用scatter函数配合对数坐标轴设置，具体修改方案如下：

修改后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt

# 创建数组
sulfate = np.array(master['SO4-2_(input)'])
chloride = np.array(master['Cl-_(input)'])
pH = np.array(master['pH'])

# 创建绘图对象
fig, ax = plt.subplots()

# 绘制带颜色映射的散点图，后续设置对数刻度
scatter = ax.scatter(chloride, sulfate, c=pH, cmap='hsv', marker='.')

# 设置坐标轴为对数刻度
ax.set_xscale('log')
ax.set_yscale('log')

# 添加1:1比例线（适配对数轴范围，避免0值导致的异常）
x_vals = np.geomspace(10.0E-7, 10.0E-1, 100)
ax.plot(x_vals, x_vals, color='gray', linestyle='--')

# 设置坐标轴范围
ax.set_xlim(10.0E-7, 10.0E-1)
ax.set_ylim(10.0E-7, 10.0E-1)

# 添加颜色图例（颜色条），标注对应的pH值
cbar = plt.colorbar(scatter)
cbar.set_label('pH')

plt.show()

关键修改说明 #

• 替换loglog为scatter：scatter函数的c参数支持传入数值数组，配合cmap参数可实现基于第三变量的颜色渐变映射，这是loglog不具备的功能。
• 手动设置对数刻度：通过ax.set_xscale('log')和ax.set_yscale('log')将坐标轴切换为对数刻度，效果和loglog一致，但保留了scatter的颜色映射能力。
• 修正1:1比例线：原代码中[0,1]在对数轴上会因log(0)无意义出现异常，改用geomspace生成对数空间的等间距数值，确保线条在有效轴范围内显示。
• 添加颜色条：通过plt.colorbar(scatter)生成对应颜色映射的图例，并设置标签说明其代表的是pH值，这就是你需要的颜色渐变图例。

内容的提问来源于stack exchange，提问作者Sien Smania