2015-11-25 12 views
6

मैं समुद्री डाकू पैकेज के साथ पाइथन में एक हीटमैप प्लॉट कर रहा हूं। जिन मूल्यों का मैं साजिश कर रहा हूं वे अलग हैं, वे पूर्णांक -1, 0, और 1 हैं।सीबर्न हीटमैप में विशिष्ट रंग के लिए मानचित्र मान

मैं हरे रंग को दिखाने के लिए -1 के साथ हीटमैप में सेल्स को पसंद करूंगा, 0 पीले रंग के रूप में, और 1 लाल के रूप में।

क्या यह निर्णय cubehelix_palette() या colour_palette() फ़ंक्शंस में निर्दिष्ट करना संभव है?

उत्तर

11

आप इस प्रकार matplotlib के ListedColormap उपयोग कर सकते हैं:

import numpy as np 
import seaborn as sns 
from matplotlib.colors import ListedColormap 

data = np.random.randint(-1, 2, (10,10)) # Random [-1, 0, 1] data 
sns.heatmap(data, cmap=ListedColormap(['green', 'yellow', 'red']), annot=True) 

जो पैदावार:

enter image description here

आप इस तरह के रूप में '#FF0000' ('red' के बराबर) hexcolors या rgb रंग के साथ तार 'green', 'yellow', 'red' जगह ले सकता है जैसे (1.,0.,0.) ('red' के बराबर)।

+0

बिल्कुल सही, धन्यवाद – jma1991

संबंधित मुद्दे