Question

我正在尝试从表格中生成一个简单的散点图，我想根据表格中的一列来对图表中的点进行颜色编码。到目前为止，我设法使用

fig = plt.figure(figsize=(14,10))
ax  = fig.add_subplot(111)
import matplotlib.pyplot as plt
ax.scatter(sub_ZH_match[:,9], sub_ZH_match[:,10], c=sub_ZH_match[:,2], s=60, marker=mod_marker, edgecolors='black', cmap=cm.jet_r, label='matches')

其中，显然

sub_ZH_match[:,9] -> x values
sub_ZH_match[:,10] -> y values
sub_ZH_match[:,2] -> values for the color coding.

但我不知道如何在情节侧面显示颜色条。

使用此语法

时有效

plt.scatter(sub_ZH_match[:,9], sub_ZH_match[:,10], c=sub_ZH_match[:,2], s=60, marker=mod_marker, edgecolors='black', cmap=cm.jet_r, label='matches')
clb = plt.colorbar()
clb.ax.tick_params(labelsize=18)
clb.set_label(r'$\nu$', fontsize=20)

但我想了解如何在坚持使用axis对象的语法时这样做。谢谢！

Answer 1

你试过这个吗？

import matplotlib.pyplot as plt
fig = plt.figure(figsize=(14,10))
ax  = fig.add_subplot(111)

scatter = ax.scatter(sub_ZH_match[:,9], sub_ZH_match[:,10], c=sub_ZH_match[:,2], s=60, marker=mod_marker, edgecolors='black', cmap=cm.jet_r, label='matches')
colorbar = fig.colorbar(scatter, ax=ax)

带有轴对象的Python散点图颜色条

1 个答案: