熊猫创建只有列名的空 DataFrame

Question

我有一个可以正常工作的动态 DataFrame，但是当没有数据要添加到 DataFrame 中时，我会收到错误消息。因此我需要一个解决方案来创建一个只有列名的空 DataFrame。

现在我有这样的事情：

 df = pd.DataFrame(columns=COLUMN_NAMES) # Note that there are now row data inserted.

PS：重要的是列名仍会出现在 DataFrame 中。

但是当我像这样使用它时，我会得到类似的结果：

 Index([], dtype='object')
Empty DataFrame

“Empty DataFrame”部分很好！但是我仍然需要显示列而不是索引。

编辑：

我发现了一件重要的事情：我正在使用 Jinja2 将此 DataFrame 转换为 PDF，因此我调用了一种方法来首先将其输出为 HTML，如下所示：

 df.to_html()

我认为这就是列丢失的地方。

Edit2：一般来说，我遵循这个例子：http: //pbpython.com/pdf-reports.html 。 CSS也来自链接。这就是我将数据框发送到 PDF 的方法：

 env = Environment(loader=FileSystemLoader('.'))
template = env.get_template("pdf_report_template.html")
template_vars = {"my_dataframe": df.to_html()}

html_out = template.render(template_vars)
HTML(string=html_out).write_pdf("my_pdf.pdf", stylesheets=["pdf_report_style.css"])

编辑3：

如果我在创建后立即打印出数据框，我会得到以下信息：

 [0 rows x 9 columns]
Empty DataFrame
Columns: [column_a, column_b, column_c, column_d,
column_e, column_f, column_g,
column_h, column_i]
Index: []

这似乎是合理的，但如果我打印出 template_vars：

 'my_dataframe': '<table border="1" class="dataframe">\n  <tbody>\n    <tr>\n      <td>Index([], dtype=\'object\')</td>\n      <td>Empty DataFrame</td>\n    </tr>\n  </tbody>\n</table>'

似乎这些列已经丢失了。

E4：如果我打印出以下内容：

 print(df.to_html())

我已经得到以下结果：

 <table border="1" class="dataframe">
  <tbody>
    <tr>
      <td>Index([], dtype='object')</td>
      <td>Empty DataFrame</td>
    </tr>
  </tbody>
</table>

原文由 E. Muuli 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以使用列名或索引创建一个空的 DataFrame：

 In [4]: import pandas as pd
In [5]: df = pd.DataFrame(columns=['A','B','C','D','E','F','G'])
In [6]: df
Out[6]:
Empty DataFrame
Columns: [A, B, C, D, E, F, G]
Index: []

或者

In [7]: df = pd.DataFrame(index=range(1,10))
In [8]: df
Out[8]:
Empty DataFrame
Columns: []
Index: [1, 2, 3, 4, 5, 6, 7, 8, 9]

编辑：即使您修改了 .to_html，我也无法重现。这个：

 df = pd.DataFrame(columns=['A','B','C','D','E','F','G'])
df.to_html('test.html')

产生：

 <table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>A</th>
      <th>B</th>
      <th>C</th>
      <th>D</th>
      <th>E</th>
      <th>F</th>
      <th>G</th>
    </tr>
  </thead>
  <tbody>
  </tbody>
</table>

原文由 Marcus V. 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

使用 iterating

 df = pd.DataFrame(columns=['colname_' + str(i) for i in range(5)])
print(df)

# Empty DataFrame
# Columns: [colname_0, colname_1, colname_2, colname_3, colname_4]
# Index: []

to_html() 操作

print(df.to_html())

# <table border="1" class="dataframe">
#   <thead>
#     <tr style="text-align: right;">
#       <th></th>
#       <th>colname_0</th>
#       <th>colname_1</th>
#       <th>colname_2</th>
#       <th>colname_3</th>
#       <th>colname_4</th>
#     </tr>
#   </thead>
#   <tbody>
#   </tbody>
# </table>

这似乎有效

print(type(df.to_html()))
# <class 'str'>

问题是由

当你像这样创建 df 时

df = pd.DataFrame(columns=COLUMN_NAMES)

它有 0 rows × n columns ，您需要至少创建一个行索引

df = pd.DataFrame(columns=COLUMN_NAMES, index=[0])

现在它有 1 rows × n columns 。您可以添加数据。否则它的 df 只包含 colnames 对象（如字符串列表）。

原文由 bitbang 发布，翻译遵循 CC BY-SA 4.0 许可协议