打印对象的漂亮打印表示的头部(前几行)

Question

我在Jupyter笔记本中运行Python 2.7。我正在使用大型嵌套词典，有时打印出其中一个会很有帮助。

使用 pprint.pprint 是在屏幕上获取可读版本的字典的好方法。但对于特别大的词典，这可能意味着打印一百万行，这会使笔记本崩溃(我认为我的浏览器无法处理它)。

在 bash 终端上，我习惯于将东西扔进| head，但在 python 中似乎没有通用的方法。

我写了这个方法：

from pprint import pformat, pprint
def pprint_head(to_print,length=10)
formatted=pformat(to_print).splitlines()
pprint(formatted[:min(len(formatted),length)])

它有效，但我想知道

1. 有没有更好/更规范/内置/"pythonic"的方法？
2. 这些小问题中的任何一个都可以改进吗？(按优先顺序)：
   • 对于大型物体来说，它很慢。
   • 它使用大量内存来处理大型对象。
   • 它被打印为字符串列表，因此它的开头有 [，每行周围都有引号。

我也想知道是否有"Jupyter"解决方案(即告诉 Jupyter 只接受任何打印的前 x 行？

Answer 1

为了获得与 shell 中的头管相同的结果，您可以轻松地在 Python 中设置输出过滤器，因为pprint只使用其流的write方法。它可能是：

class Head(object):
def __init__(self, lines, fd=sys.stdout):
self.lines = lines
self.fd = fd
def write(self, msg):
if self.lines <= 0: return
n = msg.count('n')
if n < self.lines:
self.lines -= n
return self.fd.write(msg)
ix = 0
while(self.lines > 0):
iy = msg.find('n', ix + 1)
self.lines -= 1
ix = iy
return self.fd.write(msg[:ix])

然后，您可以使用它以这种方式仅打印对象的前 n 行：

def pprint_head(to_print,length=10):
pprint(to_print, stream=Head(length))