seaborn 熱圖中的自定義 xticks

Question

我有以下熱圖（只是一個最小的工作示例，我的數據量很大！）

df = pd.DataFrame({'set1': ['0', '2', '2'],
                   'set2': ['1', '2', '0'],
                   'set3': ['0', '2', '1'],
                   'set4': ['1', '4', '1']
                  }).T.astype(float)

sns.heatmap(df, yticklabels = df.index, xticklabels = df.columns) 

我怎樣才能把 xticks 只放在那些所有rows>=2的地方？ 在此示例中，它意味着僅放置“1”xtick

因此，在此圖像中不應出現“0”和“2”列名稱，只能出現“1”，因為這是所有值都大於或等於 2 的列。

問題是 x 軸變得太擁擠了。 我只想顯示所有值 >= 2 的列的 xticklabels。仍在繪制所有內容，但只顯示那些 xticklabels。

Answer 1

您是否希望顯示相同的熱圖，但只顯示所有值 >=2 的 xticklabels？ 一種方法可能是不在熱圖中使用 df.columns，而是屏蔽並僅顯示您想要的那些。 看看這是不是你要找的……

df = pd.DataFrame({'set1': ['0', '2', '2'],
                   'set2': ['1', '2', '0'],
                   'set3': ['0', '2', '1'],
                   'set4': ['1', '4', '1']
                  }).T.astype(float)

cols = [] ## Declare a new list to be used for xticklabels
for col in df.columns:
    if col in set(df.columns).intersection(df[df >= 2].T.dropna().index):
        cols.append(col) ##If all values are >=2, then add to label list
    else:
        cols.append('') ## Else, add blank

sns.heatmap(df, yticklabels = df.index, xticklabels = cols) ##Plot using new list

Answer 2

掩碼 DataFrame

• 刪除值不>=到指定值的列

# create a Boolean mask of df
mask = df.ge(2)

# apply the mask to df and dropna
dfm = df[mask].dropna(axis=1)

# plot the masked df
ax = sns.heatmap(dfm)

mask

          0     1      2
set1  False  True   True
set2  False  True  False
set3  False  True  False
set4  False  True  False

屏蔽 xtick 標簽

• 值不>=到指定值的列的標簽將替換為'' 。

# create a Boolean mask of df
mask = df.ge(2).all()

# use the mask to update a list of labels
cols = [col if m else '' for (col, m) in zip(df.columns, mask)]

# plot with custom labels
ax = sns.heatmap(df, xticklabels=cols)

mask

0    False
1     True
2    False
dtype: bool