在Python Pandas DataFrame或Jupyter笔记本中包装列名

Question

我的数据框中有一些列的长标题,我希望能够包装文本.我知道这个功能内置于熊猫中,就像我一样:

pd.DataFrame(np.random.randn(2, 10), 
    columns=['Very Long Column Title ' + str(i) for i in range(10)])

包含列名的DataFrame

但是如果我有更少的列,标题将不会换行:

pd.DataFrame(np.random.randn(10, 2), 
    columns=['Very Long Column Title ' + str(i) for i in range(2)])

DataFrame不包装列名

我还尝试手动插入换行符:

import pandas as pd    
pd.DataFrame(np.random.randn(10, 2), 
    columns=['Very Long \n Column Title ' + str(i) for i in range(2)])

但是,它提供与上面相同的输出.

我发现类似的答案有关这个主题:

• 我可以在pandas中设置可变列宽吗？
  将截断列宽,但不会影响标题,也不会包装文本
• Pandas DataFrame中字符串中的漂亮打印换行符
  再次触及列内容但不触及标题

我正在使用Jupyter笔记本电脑,但如果可能的话,我更喜欢基于熊猫的解决方案.

Answer 1

Jupyter笔记本从许多来源继承了它们的显示属性.没有属性pandas限制列标题的宽度,因为pandas它不是导致文本换行的原因,它实际上是呈现的HTML.

您可以使用以下方法覆盖默认的Jupyter Notebook样式以限制表头的最大宽度:

from IPython.core.display import HTML
HTML("<style>.rendered_html th {max-width: 120px;}</style>")

在笔记本顶部运行此代码一次,将html表头的最大列宽设置为120像素.

Answer 2

这是一个不涉及更改IPython属性的答案：

df = pd.DataFrame(np.random.randn(10, 2), 
    columns=['Very Long Column Title ' + str(i) for i in range(2)])
df.style.set_table_styles([dict(selector="th",props=[('max-width', '50px')])])