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)