Pandas：DataFrame 和 Series 的通用列属性

Question

我有一个函数，它通过 DataFrame 并使用 df.columns 属性。但是，当它传递一个只有一列（系列）的 DataFrame 时，这会中断，因为系列没有该属性。

是否有一个属性适用于系列和数据框，相当于 df.columns 的 DataFrame ？我看过了，但鉴于列属性甚至没有显示在 DataFrame 文档 (http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.html) 中，我认为可能有一些部落知识形式的解决方案。

Answer 1

将.name 用于系列，我只需测试被传递对象的type 并酌情调用.columns 或.name：

In [242]:
df = pd.DataFrame(columns=list('abc'))
df

Out[242]:
Empty DataFrame
Columns: [a, b, c]
Index: []

In [243]:
df['a'].name

Out[243]:
'a'

像这样：

In [245]:
def func(x):
    if type(x) is pd.DataFrame:
        print( x.columns)
    elif type(x) is pd.Series:
        print( x.name)

print(func(df))
print(func(df['a']))

Index(['a', 'b', 'c'], dtype='object')
None
a
None