從熊貓列索引中提取列名稱

Question

我有一個函數，它需要在散點圖上繪製Pandas列。我想使用列的名稱作爲軸標籤。下面是我的意思的一個例子：

def plotscatter(x,y): 
    plt.scatter(x, y) 
    plt.xlabel(x) 
    plt.ylabel(y) 
plotscatter(df['1stcol'],df['2ndcol']) 

我想只提取引號中的文本，即1stcol和2ndcol。如果我按照原樣使用x和y，我會得到列名，然後是所有值作爲軸標籤。

有沒有一個內置的方法或函數呢？像x.columnname（）

Answer 1

您可以通過DataFrame.columns或列Series.name訪問列名稱。舉個例子：

In [11]: df = pd.DataFrame([[1,2],[2,4]], columns=['1stcol', '2ndcol']) 

In [13]: df.columns 
Out[13]: Index([u'1stcol', u'2ndcol'], dtype='object') 

In [14]: df.columns[0] 
Out[14]: '1stcol' 

In [16]: df['1stcol'].name 
Out[16]: '1stcol'