[英]How do I find color limits for qualitative colormaps in Matplotlib?
我有以下代码生成散点图 plot:
plt.scatter(df['column_A'], y= df['column_B'],
alpha = .6, c = df['column_C'], cmap = 'Accent')
我在 cmap 中使用定性颜色图,它会自动在 column_C 上创建间隔并相应地为点着色。 我可以直观地说灰色组的范围从大约 10 到数据中的最大值(大约 12),但我想访问确切的数字。 有没有办法做到这一点?
非常感谢
默认情况下,colors 的范围从column_C
的最小值到最大值。 对于具有N
colors 的颜色图,将有N
个段,因此可以使用np.linspace()
计算N+1
边界。
这是一些说明性代码。 添加了一个带有默认刻度的颜色条和一个在颜色边界处带有刻度的颜色条。
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
df = pd.DataFrame({'column_A': np.random.rand(100),
'column_B': np.random.rand(100),
'column_C': np.random.rand(100) * 20 - 8})
cmap = plt.get_cmap('Accent')
plt.scatter(df['column_A'], y=df['column_B'],
alpha=.6, c=df['column_C'], cmap=cmap)
plt.colorbar() # the default color bar
color_borders = np.linspace(df['column_C'].min(), df['column_C'].max(), cmap.N + 1)
plt.colorbar(ticks=color_borders) # ticks at the color borders
plt.show()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.