pandas - 如何在style.apply之后显示 Pandas 数据框的截断形式？

Question

通常，一个相对较长的数据帧，如

df = pd.DataFrame(np.random.randint(0,10,(100,2)))
df

将在 jupyter notebook 中显示一个截断的形式，如



头、尾、省略号在中间，行列数在最后。
但是，在 style.apply 之后

def highlight_max(x):
    return ['background-color: yellow' if v == x.max() else '' for v in x]
df.style.apply(highlight_max)

我们显示了所有行



在style.apply之后是否仍然可以显示数据框的截断形式？

Answer 1

这么简单的事？

def display_df(dataframe, function):
    display(dataframe.head().style.apply(function))
    display(dataframe.tail().style.apply(function))
    print(f'{dataframe.shape[0]} rows x {dataframe.shape[1]} columns')

display_df(df, highlight_max)

输出:



**** 编辑 ****

def display_df(dataframe, function):
    display(pd.concat([dataframe.iloc[:5,:],
                       pd.DataFrame(index=['...'], columns=dataframe.columns),
                       dataframe.iloc[-5:,:]]).style.apply(function))
    print(f'{dataframe.shape[0]} rows x {dataframe.shape[1]} columns')

display_df(df, highlight_max)

输出:



jupyter 预览基本上是这样的:

def display_df(dataframe):
    display(pd.concat([dataframe.iloc[:5,:],
                       pd.DataFrame(index=['...'], columns=dataframe.columns, data={0: '...', 1: '...'}),
                       dataframe.iloc[-5:,:]]))

但是，如果您尝试应用样式，则会出现错误(TypeError: '>=' not supported between 'int' 和 'str')，因为它正在尝试比较和突出显示字符串值 '...'