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

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

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

df.to_html()

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

一般来说，我遵循这个例子： http://pbpython.com/pdf-reports.html 。 css 也来自链接。 这就是我将 dataframe 发送到 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"])

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>

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')

Answer 3

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

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

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

Answer 4

通过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 对象（如字符串列表）。

Answer 5

您可以使用pd.DataFrame() function 来执行此操作，并将列名称传递为 arguments. like。

 df = pd.DataFrame(columns=['col1','col2','col3'])

很高兴能帮助你！

Answer 6

如果你有一个没有列或索引的完全空的 dataframe，你可以通过将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)