python - Pandas 创建只有列名的空 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()

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

编辑2: 一般来说，我遵循这个例子: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>

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>