147

我有一个从 Python 的 Pandas 包生成的数据框。如何使用 pandas 包中的 DataFrame 生成热图。

import numpy as np 
from pandas import *

Index= ['aaa','bbb','ccc','ddd','eee']
Cols = ['A', 'B', 'C','D']
df = DataFrame(abs(np.random.randn(5, 4)), index= Index, columns=Cols)

>>> df
          A         B         C         D
aaa  2.431645  1.248688  0.267648  0.613826
bbb  0.809296  1.671020  1.564420  0.347662
ccc  1.501939  1.126518  0.702019  1.596048
ddd  0.137160  0.147368  1.504663  0.202822
eee  0.134540  3.708104  0.309097  1.641090
>>> 
4

7 回答 7

232

对于今天看到这个的人,我会推荐这里heatmap()记录的 Seaborn 。

上面的示例将按如下方式完成:

import numpy as np 
from pandas import DataFrame
import seaborn as sns
%matplotlib inline

Index= ['aaa', 'bbb', 'ccc', 'ddd', 'eee']
Cols = ['A', 'B', 'C', 'D']
df = DataFrame(abs(np.random.randn(5, 4)), index=Index, columns=Cols)

sns.heatmap(df, annot=True)

%matplotlib对于那些不熟悉的人来说,IPython 魔术函数在哪里。

于 2015-04-09T02:00:13.673 回答
133

如果您不需要一个图,并且您只是对添加颜色以表示表格格式的值感兴趣,您可以使用style.background_gradient()pandas 数据框的方法。此方法对在例如 JupyterLab Notebook 中查看 pandas 数据帧时显示的 HTML 表格进行着色,结果类似于在电子表格软件中使用“条件格式”:

import numpy as np 
import pandas as pd


index= ['aaa', 'bbb', 'ccc', 'ddd', 'eee']
cols = ['A', 'B', 'C', 'D']
df = pd.DataFrame(abs(np.random.randn(5, 4)), index=index, columns=cols)
df.style.background_gradient(cmap='Blues')

在此处输入图像描述

有关详细用法,请参阅我之前在同一主题上提供的更详细的答案以及pandas 文档的样式部分

于 2018-05-30T12:43:28.707 回答
101

你想要matplotlib.pcolor

import numpy as np 
from pandas import DataFrame
import matplotlib.pyplot as plt

index = ['aaa', 'bbb', 'ccc', 'ddd', 'eee']
columns = ['A', 'B', 'C', 'D']
df = DataFrame(abs(np.random.randn(5, 4)), index=index, columns=columns)

plt.pcolor(df)
plt.yticks(np.arange(0.5, len(df.index), 1), df.index)
plt.xticks(np.arange(0.5, len(df.columns), 1), df.columns)
plt.show()

这给出了:

输出样本

于 2012-09-05T17:42:37.583 回答
21

有用sns.heatmap的 api 在这里。查看参数,其中有很多。例子:

import seaborn as sns
%matplotlib inline

idx= ['aaa','bbb','ccc','ddd','eee']
cols = list('ABCD')
df = DataFrame(abs(np.random.randn(5,4)), index=idx, columns=cols)

# _r reverses the normal order of the color map 'RdYlGn'
sns.heatmap(df, cmap='RdYlGn_r', linewidths=0.5, annot=True)

在此处输入图像描述

于 2017-05-17T19:46:23.810 回答
6

如果您想要来自 Pandas DataFrame 的交互式热图并且您正在运行 Jupyter 笔记本,您可以尝试交互式 Widget Clustergrammer-Widget ,请参阅此处的 NBViewer 上的交互式笔记本,此处的文档

在此处输入图像描述

对于更大的数据集,您可以尝试开发中的Clustergrammer2 WebGL 小部件(此处的示例笔记本)

于 2019-03-27T15:44:40.953 回答
4

请注意,作者seaborn只想使用分类数据框 seaborn.heatmap这不是一般的。

如果您的索引和列是数字和/或日期时间值,则此代码将为您提供很好的服务。

Matplotlib 热映射功能pcolormesh需要bin而不是indices,因此有一些花哨的代码可以从您的数据帧索引构建 bin(即使您的索引不是均匀分布的!)。

剩下的就是np.meshgridplt.pcolormesh

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

def conv_index_to_bins(index):
    """Calculate bins to contain the index values.
    The start and end bin boundaries are linearly extrapolated from 
    the two first and last values. The middle bin boundaries are 
    midpoints.

    Example 1: [0, 1] -> [-0.5, 0.5, 1.5]
    Example 2: [0, 1, 4] -> [-0.5, 0.5, 2.5, 5.5]
    Example 3: [4, 1, 0] -> [5.5, 2.5, 0.5, -0.5]"""
    assert index.is_monotonic_increasing or index.is_monotonic_decreasing

    # the beginning and end values are guessed from first and last two
    start = index[0] - (index[1]-index[0])/2
    end = index[-1] + (index[-1]-index[-2])/2

    # the middle values are the midpoints
    middle = pd.DataFrame({'m1': index[:-1], 'p1': index[1:]})
    middle = middle['m1'] + (middle['p1']-middle['m1'])/2

    if isinstance(index, pd.DatetimeIndex):
        idx = pd.DatetimeIndex(middle).union([start,end])
    elif isinstance(index, (pd.Float64Index,pd.RangeIndex,pd.Int64Index)):
        idx = pd.Float64Index(middle).union([start,end])
    else:
        print('Warning: guessing what to do with index type %s' % 
              type(index))
        idx = pd.Float64Index(middle).union([start,end])

    return idx.sort_values(ascending=index.is_monotonic_increasing)

def calc_df_mesh(df):
    """Calculate the two-dimensional bins to hold the index and 
    column values."""
    return np.meshgrid(conv_index_to_bins(df.index),
                       conv_index_to_bins(df.columns))

def heatmap(df):
    """Plot a heatmap of the dataframe values using the index and 
    columns"""
    X,Y = calc_df_mesh(df)
    c = plt.pcolormesh(X, Y, df.values.T)
    plt.colorbar(c)

使用 调用它heatmap(df),并使用 来查看它plt.show()

在此处输入图像描述

于 2019-07-01T18:58:31.747 回答
4

惊讶地发现没有人提到更强大、更具交互性和更易于使用的替代方案。

A)你可以使用情节:

  1. 只需两行,您就会得到:

  2. 互动性,

  3. 平滑的刻度,

  4. 基于整个数据框而不是单个列的颜色,

  5. 轴上的列名和行索引,

  6. 放大,

  7. 平移,

  8. 内置一键式功能将其保存为 PNG 格式,

  9. 自动缩放,

  10. 悬停比较,

  11. 显示值的气泡,因此热图仍然看起来不错,您可以在任何地方看到值:

import plotly.express as px
fig = px.imshow(df.corr())
fig.show()

在此处输入图像描述

B)您还可以使用散景:

所有相同的功能都有一点麻烦。但是,如果您不想选择加入 plotly 并且仍然想要所有这些东西,那么仍然值得:

from bokeh.plotting import figure, show, output_notebook
from bokeh.models import ColumnDataSource, LinearColorMapper
from bokeh.transform import transform
output_notebook()
colors = ['#d7191c', '#fdae61', '#ffffbf', '#a6d96a', '#1a9641']
TOOLS = "hover,save,pan,box_zoom,reset,wheel_zoom"
data = df.corr().stack().rename("value").reset_index()
p = figure(x_range=list(df.columns), y_range=list(df.index), tools=TOOLS, toolbar_location='below',
           tooltips=[('Row, Column', '@level_0 x @level_1'), ('value', '@value')], height = 500, width = 500)

p.rect(x="level_1", y="level_0", width=1, height=1,
       source=data,
       fill_color={'field': 'value', 'transform': LinearColorMapper(palette=colors, low=data.value.min(), high=data.value.max())},
       line_color=None)
color_bar = ColorBar(color_mapper=LinearColorMapper(palette=colors, low=data.value.min(), high=data.value.max()), major_label_text_font_size="7px",
                     ticker=BasicTicker(desired_num_ticks=len(colors)),
                     formatter=PrintfTickFormatter(format="%f"),
                     label_standoff=6, border_line_color=None, location=(0, 0))
p.add_layout(color_bar, 'right')

show(p)

在此处输入图像描述

于 2020-11-29T01:10:57.560 回答