Python next() 用于 5 个示例并存储在 Dataframe 中

Question

我有一个文本数据集，我想将前 5 行可视化到一个数据框中。有没有一种优雅的方法可以从 next() 打印 5 次迭代？

from torchtext.datasets import Multi30k
train_data, valid_data, test_data = Multi30k(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'))

next(train_data)  # How do I print 5 "next" in a loop and send it to a dataframe?

第二部分，我如何转置元组，使其成为两列？例如;元组的第一部分是列 'German'，第二部分是 'English'。这是我尝试过的：

line1=next(train_data)
pd.DataFrame(list(line1))

但这只是 returns 具有一列和两行的数据框。我希望目标是这样的图像：

Answer 1

您可以使用 itertools islice 来实现。

from torchtext.datasets import Multi30k
from itertools import islice
import pandas as pd

train_data, valid_data, test_data = Multi30k(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'))

pd.DataFrame(list(islice(train_data,5)), columns=['de','en'])

Python next() 用于 5 个示例并存储在 Dataframe 中

Python next() for 5 examples and store in a Dataframe

python

pandas

next