python - 有没有办法将 Pandas 数据框打印为可复制代码？

Question

我时不时地需要调试代码，然后找到一个有效的“oracle”或“fixture”用于新的测试用例，然后需要将数据帧打印到控制台并将其转换为数据帧初始化代码。有没有办法将数据框打印成可复制的代码？我知道我可以将它转储到一个文件中，然后使用该文件作为测试用例的输入，对于很长的数据帧应该是这种情况，但对于小数据帧来说，能够将它们复制粘贴到一个测试套件...

Answer 1

如果您寻求的是可以用作有效 Python 代码的 DataFrame 表示，建议您使用 to_dict ， 如下:

import pandas as pd

df = pd.DataFrame(data=list(enumerate(['Dan Cat', 'Dave Dog', 'Diego Cangrejo'], 1)), columns=['id', 'name'])
print(df.to_dict())

输出

{'id': {0: 1, 1: 2, 2: 3}, 'name': {0: 'Dan Cat', 1: 'Dave Dog', 2: 'Diego Cangrejo'}}

上述状态的输出是有效的 Python 字典字面量，因此您可以直接在代码中复制它:

import pandas as pd

data = {'id': {0: 1, 1: 2, 2: 3}, 'name': {0: 'Dan Cat', 1: 'Dave Dog', 2: 'Diego Cangrejo'}}
print(pd.DataFrame.from_dict(data))

输出

   id            name
0   1         Dan Cat
1   2        Dave Dog
2   3  Diego Cangrejo