如何制作不同大小,颜色和位置的散点图?

时间:2015-09-22 14:46:59

标签: python matplotlib scatter

如何使用delta_x和delta_y位置的随机值制作散点图;其中每个点具有特定的频率值(颜色强度根据强度而变化),即某些符号。

示例情节:(来自Alberdi, et al, 2013

enter image description here

1 个答案:

答案 0 :(得分:1)

如果我正确理解你,你问的是如何让scatter分享一个色标,但对不同的群体有不同的符号,对吗?

有几种不同的方法可以解决这个问题。

关键是多次调用scatter(每个不同的组一个),但传入相同的vminvmaxcmap个参数。

作为复制上述情节的完整(并且可以说是过于复杂)的例子:

import numpy as np
import matplotlib.pyplot as plt

# Generate data
freq_groups = [1.7, 2.3, 5.0, 8.4]
num = 50
x = np.random.normal(0, 0.5, num)
y = np.random.normal(0.2, 0.5, num)
year = 9 * np.random.random(num) + 1993.5
frequencies = np.random.choice(freq_groups, num)

symbols = ['o', '^', 's', 'd']

# Plot data
fig, ax = plt.subplots(figsize=(8, 9))
for freq, marker in zip(freq_groups, symbols):
    mask = np.isclose(freq, frequencies)
    scat = ax.scatter(x[mask], y[mask], c=year[mask], s=100, marker=marker,
                      cmap='jet_r', vmin=year.min(), vmax=year.max(),
                      label='{:0.1f} GHz'.format(freq), color='black')

ax.legend(loc='upper left', scatterpoints=1)
ax.set(xlabel='Relative RA (mas)', ylabel='Relative Dec (mas)')
ax.invert_xaxis()
cbar = fig.colorbar(scat, orientation='horizontal')
cbar.set_label('Epoch (year)')
cbar.formatter.useOffset = False
cbar.update_ticks()
fig.tight_layout()

plt.show()

enter image description here