从 csv 文件提供示例 DataFrame [重复]

Question

在 stackoverflow 上询问 python/pandas 问题时，我经常喜欢提供示例数据框。 我通常有一个本地 csv 文件用于测试。

因此，对于 DataFrame，我喜欢在我的问题中提供代码，例如

df = pd.DataFrame()

有没有一种简单的方法或工具可以将 csv 文件转换为这样的格式的代码，以便其他用户可以轻松地重新创建数据框？

目前我通常手动完成，这很烦人且耗时。我必须将数据从 Excel 复制/粘贴到 stackoverflow、删除制表符/空格、重新排列数字以获取列表或字典等等。

csv 文件示例：

col1	col2
1	3
2	4

如果我想提供这个表格，我可以提供如下代码：

d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)

我必须手动创建字典和数据框。我必须手动将代码写入 stackoverflow 编辑器。 对于更复杂的表，这可能会导致大量工作。

希望你能解决“问题”。

Answer 1

您可以从 dict

创建一个 .csv 并将其传递给

DataFrame

构造函数:

N = 5  # <- adjust here to choose the number of rows

dico = pd.read_csv("f.csv").sample(N).to_dict("list")

S = f"df = pd.DataFrame({dico})"
print(S)  # <- copy its output and paste in Stack Overflow

您还可以使用

pyperclip

直接复制要粘贴/包含在问题中的文本：

#pip install pyperclip
import pyperclip

pyperclip.copy(S)