如何在Matplotlib坐标轴间正确迁移Collections？

遇到这个问题太正常了——毕竟Matplotlib里的Collection（比如plt.scatter生成的PathCollection）和普通Artist（比如线条、文本）的坐标处理逻辑完全不一样，只改transform肯定解决不了根本问题。我来一步步给你拆解解决方案：

问题根源 #

普通Artist（比如Line2D）的位置是通过transform属性映射到画布的，切换transform就能直接把它的坐标从原轴转到目标轴。但Collection不一样：它的点位置存在offsets属性里，这些是原坐标轴的原始数据坐标值，不会自动跟着transform变化。你之前移除set_transform能有改善，是因为add_collection会自动把Collection的transform设为目标轴的transData，但offsets还是原轴的数据，自然会出现偏移。

正确操作步骤 #

核心是把offsets从原轴的数据坐标，转换为目标轴的数据坐标。具体操作如下：

1. 从原轴移除Collection
   先把要迁移的Collection从原坐标轴的collections列表里删掉：

   ax1.collections.remove(scatter_collection)
   

2. 转换offsets的坐标系统
   我们需要把原offsets先转成画布的显示坐标（display coordinates），再转回目标轴的数据坐标：

   # 获取原offsets（原轴的data坐标）
   original_offsets = scatter_collection.get_offsets()
   # 转成画布显示坐标
   display_coords = ax1.transData.transform(original_offsets)
   # 转回目标轴的data坐标
   new_offsets = ax2.transData.inverted().transform(display_coords)
   # 更新Collection的offsets
   scatter_collection.set_offsets(new_offsets)
   

3. 添加到目标轴并调整显示
   把处理好的Collection添加到目标轴，然后触发自动缩放确保所有点都在视野里：

   ax2.add_collection(scatter_collection)
   ax2.autoscale()
   

完整示例代码 #

import matplotlib.pyplot as plt
import numpy as np

# 创建两个坐标轴
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(6, 8))

# 在ax1生成scatter的PathCollection
x = np.random.rand(50)
y = np.random.rand(50)
sc = ax1.scatter(x, y, color='teal', s=100)
ax1.set_title('Original Axis')

# 开始迁移操作
ax1.collections.remove(sc)

# 转换offsets坐标
original_offsets = sc.get_offsets()
display_coords = ax1.transData.transform(original_offsets)
new_offsets = ax2.transData.inverted().transform(display_coords)
sc.set_offsets(new_offsets)

# 添加到ax2并调整
ax2.add_collection(sc)
ax2.autoscale()
ax2.set_title('Migrated Axis')

plt.tight_layout()
plt.show()

额外注意事项 #

• 如果你的Collection使用了颜色映射（比如基于数据值的颜色），需要确保norm的范围和目标轴的数据源匹配，否则颜色显示可能异常。
• 对于其他类型的Collection（比如LineCollection），逻辑类似：需要把对应的坐标数据（比如segments）做同样的坐标转换。

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