Pandas仅使用列名创建空DataFrame

时间:2017-06-13 06:22:58

标签: python pandas dataframe

我有一个动态的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,因此我调用了一种方法,首先将其输出为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"])

EDIT3:

如果我在创建后立即打印出数据框,我会得到以下内容:

[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>

4 个答案:

答案 0 :(得分:113)

您可以使用列名或索引创建空的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>

答案 1 :(得分:5)

你在找这样的东西吗?

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

答案 2 :(得分:1)

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

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

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

答案 3 :(得分:1)

使用 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 对象(如字符串列表)。