Pandas创建只有列名的空DataFrame

Question

我有一个动态数据框，它可以正常运行，但当没有数据添加到数据框中时，我会收到一个错误。因此，我需要一个解决方案来创建一个仅包含列名的空数据框。

目前我有类似以下代码：

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

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

但是当我像这样使用它时，我得到了以下结果：

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

“空数据框”部分很好！但是我仍然需要显示列，而不是索引。

我发现一个重要的事情：我正在使用Jinja2将此DataFrame转换为PDF，因此我首先调用一个方法将其输出到HTML：

df.to_html()

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

总体上，我遵循了这个例子：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"])

- E. Muuli

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.

Answer 2

你是否在寻找类似这样的东西？

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

   Index(['A', 'B', 'C', 'D', 'E', 'F', 'G'], dtype='object')

- Linda

---

https://dev59.com/0FwX5IYBdhLWcg3w-Thv - Linda

此外，当我尝试时，我的列名并没有丢失。它以HTML表格格式呈现。 - Linda

Answer 3

使用迭代创建列名

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行×n列，您需要通过创建至少一个行索引来解决此问题。

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

现在它有1行×n列。您可以添加数据。否则，它的df仅包含colnames对象(类似于字符串列表)。

- bitbang

---

非常感谢您。我曾经苦苦挣扎了两个小时，无法添加数据。 - Mikk

Answer 4

df.to_html()方法有一个columns参数。

只需将列名传递给to_html()方法即可。

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

- Eric

Answer 5

如果您有一个完全没有列或索引的空数据框，您可以通过将None分配给这些列来让它具有列。

df = pd.DataFrame()                    # <---- shape: (0, 0)
df[['col1', 'col2', 'col3']] = None    # <---- shape: (0, 3)

然后要将一行分配给它，您可以使用loc索引器。实际上，这可以在循环中使用以添加更多行（不建议这样做，因为pd.concat存在于执行该特定任务）。

df.loc[len(df)] = ['abc', 10, 3.33]    # <---- shape: (1, 3)

- cottontail