Setting columns for an empty pandas dataframe

Question

This is something that I\'m confused about...

import pandas as pd

# this works fine
df1 = pd.DataFrame(columns=[\'A\',\'B\'])

# but let\'s say I have this
d         


        

Answer 1

This looks like a bug in pandas. All of these work:

pd.DataFrame(columns=['A', 'B'])
pd.DataFrame({}, columns=['A', 'B'])
pd.DataFrame(None, columns=['A', 'B'])

but not this:

pd.DataFrame([], columns=['A', 'B'])

Until it's fixed, I suggest something like this:

if len(data) == 0: data = None
df2 = pd.DataFrame(data, columns=['A','B'])

or:

df2 = pd.DataFrame(data if len(data) > 0 else None, columns=['A', 'B'])