從 Matplotlib 中的散點圖中提取數據

Question

我正在編寫一個接口來在 Matplotlib 中繪制散點圖，我希望能夠從 python 腳本訪問數據。

現在，我的界面正在做：

scat = self.axes.scatter(x_data, y_data, label=label, s=size)

使用標准的axes.plot我可以執行以下操作：

line = self.axes.plot(x_data, y_data)
data = line[0].get_data()

這有效。 我想要的是類似的東西，但有散點圖。

有人可以建議類似的方法嗎？

Answer 1

一個scatter曲線使用繪制PathCollection ，所以X，Y位置被稱為“偏移”：

import numpy as np
import matplotlib.pyplot as plt

f, ax = plt.subplots()
scat = ax.scatter(np.random.randn(10), np.random.randn(10))

print scat.get_offsets()

[[-0.17477838 -0.47777312]
 [-0.97296068 -0.98685982]
 [-0.18880346  1.16780445]
 [-1.65280361  0.2182109 ]
 [ 0.92655599 -1.40315507]
 [-0.10468029  0.82269317]
 [-0.09516654 -0.80651275]
 [ 0.01400393 -1.1474178 ]
 [ 1.6800925   0.16243422]
 [-1.91496598 -2.12578586]]